分类:全部列表
问题描述:查看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...
继续阅读 >
1squeeze():去除size为1的维度,包括行和列。至于维度大于等于2时,squeeze()不起作用。行、例:>>>torch.rand(4,1,3)(0,.,.)=0.53910.85230.9260(1,.,.)=0.25070.95120.6578(2,.,.)=0.73020.35310.9442(3,.,.)=0.26890.43670.6610[torch.FloatTensorofsize4x1x3]>>>torch.rand(4,1,3).squeeze()0.08010.46000.17990.02360.71370.61280.02420.38470.45460.9004...
继续阅读 >