torch.Tensor().flip()
torch.Tensor().flip(dim)
作用
指定dim反转张量
示例
a = torch.Tensor([1,0])
print(a.flip(0))
>>> tensor([0., 1.])
指定dim反转张量
a = torch.Tensor([1,0])
print(a.flip(0))
>>> tensor([0., 1.])
转载于:https://www.cnblogs.com/Zak-NoS/p/10975014.html