torch 中的Tensor index和numpy是类似的,参看下列代码输出 ... r = torch.squeeze(t, 1) # Squeeze dimension 1: Size 2x2x1.
確定! 回上一頁