qq_43581151的博客

改变Tensor尺寸的操作1.tensor.viewtensor.view方法，可以调整tensor的形状，但必须保证调整前后元素总数一致。view不会改变自身数据，返回的新的tensor与源tensor共享内存，即更改其中一个，另外一个也会跟着改变。例：In： import torch as t a = t.arange(0, 6) a.view(2, 3)Out：...

1.gather函数gather(input, dim, index)：根据index，在dim维度上选取数据，输出的size与index一样。gather是一个比较复杂的操作，对一个二维tensor，输出的每个元素如下：out[i][j] = input[index[i][j]][j] #dim = 0out[i][j] = input[i][index[i][j]] #dim ...

Tensor有不同的数据类型，如下表所示，每种类型分别有CPU版本和GPU版本（HalfTensor除外）。默认的tensor类型是FloatTensor，可通过t.set_default_tensor_type修改默认tensor类型（如果默认类型为GPU tensor，则所有的操作都在GPU上进行）。数据类型CPU tensorGPU tensor32bit浮点tor...