Pytorch | 接口［１］

2024年3月23日 199次阅读来源: pytorch

torch.Tensor().flip()

　　指定dim反转张量

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

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

    原文作者：pytorch
    原文地址: https://www.cnblogs.com/Zak-NoS/p/10975014.html
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。