分类：数据类型大小

2021
07-08

pytorch 如何查看数据类型和大小

问题描述：查看tensor数据大小时使用了data.shape(),报错：TypeError:'torch.Size'objectisnotcallable或TypeError:'tuple'objectisnotcallable。解决方法：查看数据类型：data.dtype查看数据大小：data.shape补充：pytorchtensor比较大小数据类型要注意如下a=torch.tensor([[0,0],[0,0]])print(a>=0.5)输出tensor([[1,1],[1,1]],dtype=torch.uint8)结果明显不对,分析原因是因为,a是long类型,而0... 继续阅读 >