pytorch中index select 的用法

a = torch.linspace(1, 12, steps=12).view(3, 4)

print
(a)b = torch.index_select(a, 0, torch.tensor([0, 2]))
print
(b)print(a.index_select(0, torch.tensor([0, 2])))
c = torch.index_select(a, 1, torch.tensor([1, 3]))
print(c)

先定義了乙個tensor，這裡用到了linspace和view方法。

第乙個引數是索引的物件，第二個引數0表示按行索引，1表示按列進行索引，第三個引數是乙個tensor，就是索引的序號，比如b裡面tensor[0， 2]表示第0行和第2行，c裡面tensor[1, 3]表示第1列和第3列。

輸出結果如下：

tensor([[ 1.,  2.,  3.,  4.],

[ 5.,  6.,  7.,  8.],
[ 9., 10., 11., 12.]])
tensor([[ 1.,  2.,  3.,  4.],
[ 9., 10., 11., 12.]])
tensor([[ 1.,  2.,  3.,  4.],
[ 9., 10., 11., 12.]])
tensor([[ 2.,  4.],
[ 6.,  8.],
[10., 12.]])

Pytorch 中 torchvision的錯誤

在學習pytorch的時候，使用 torchvision的時候發生了乙個小小的問題 安裝都成功了，並且import torch也沒問題，但是在import torchvision的時候，出現了如下所示的錯誤資訊 dll load failed 找不到指定模組。首先，我們得知道torchvision在...

Pytorch中建立DataLoader的幾種方法

簡介 這段 是mnist手寫體識別中的部分 此篇 為mnist手寫體識別中的 import torch import torchvision import torchvision.transforms as transforms from torch.utils.data import datalo...

pytorch中的乘法

總結 按元素相乘用torch.mul，二維矩陣乘法用torch.mm，batch二維矩陣用torch.bmm，batch 廣播用torch.matmul if name main a torch.tensor 1 2,3 b torch.arange 0,12 reshape 4 3 c torch...