随笔分类 -  python

expected device cpu and dtype Float but got device cpu and dtype Long
摘要：再练习torch时遇到的bug 将两个tensor相加，报错，不能相加 a = torch.arange(16).reshape((4, 4)) a tensor([[ 0, 1, 2, 3], [ 4, 5, 6, 7], [ 8, 9, 10, 11], [12, 13, 14, 15]]) b 阅读全文
posted @ 2021-01-17 18:54 zpcode 阅读(57) 评论(0) 推荐(0)