Pytorch | 接口［１］

torch.Tensor().flip()

torch.Tensor().flip(dim)

作用

　　指定dim反转张量

示例

a = torch.Tensor([1,0])
print(a.flip(0))

>>> tensor([0., 1.])