我正试着习惯于这种情况.然而,我无法理解张量[:,-1](应该打印最后一列)和张量[…,-1]之间的区别,后者打印不同的输出(output2)
import torch
tensor = torch.rand([3,3,3,3])
print('Output1')
print(tensor[:,-1])
print('Output2')
print(tensor[...,-1])
我正试着习惯于这种情况.然而,我无法理解张量[:,-1](应该打印最后一列)和张量[…,-1]之间的区别,后者打印不同的输出(output2)
import torch
tensor = torch.rand([3,3,3,3])
print('Output1')
print(tensor[:,-1])
print('Output2')
print(tensor[...,-1])
看起来以下指标是等效的
tensor[:, -1] == tensor[:, -1, :, :]
tensor[..., -1] == tensor[:, :, :, -1]