说明
最近在使用pytorch中tensor的时候,遇到了需要使用tensor1[tensor2]的情况,发现了这篇博客,他从代码的角度解释了其工作原理,这里我用图示的方式解释tensor类型为longTensor时的情况
工作原理
- 首先,创建两个tensor,如下:
import numpy as np import torch x1 = np.array([[1,0,0,1],
最近在使用pytorch中tensor的时候,遇到了需要使用tensor1[tensor2]的情况,发现了这篇博客,他从代码的角度解释了其工作原理,这里我用图示的方式解释tensor类型为longTensor时的情况
import numpy as np
import torch
x1 = np.array([[1,0,0,1],