pytorch将小数tensor转换成整数tensor

a = [[1.,2.],[3.,4.]]
b = torch.tensor(a)
# c = torch.tensor(b,dtype=torch.int)
c = b.clone().type(torch.int)

print(b)
print(c)
    原文作者:咖乐布小部
    原文地址: https://blog.csdn.net/weixin_42253689/article/details/120298269
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞