TypeError: Cannot interpret ‘2‘ as a data type

错误行：

查看ones 定义

所以应该是:

b = torch.as_tensor(np,ones((2,3)))    这里省略了dtype.。

 

 

版权声明：本文为weixin_42253689原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。