選取部分張量元素

代碼1: t=torch.arange(12) print(t) print(t[3],t[0]) print(t[-5]) print(t[3:6])#個數爲3 print(t[:6]) print(t[-5:]) print(t[3:6:2])#2表示只有2個元素 print(t[3::2])#以2位方差  運行結果: 代碼2 :   t= torch.arange(24).reshape(2
相關文章
相關標籤/搜索