AttributeError: ‘torch.return_types.max‘ object has no attribute ‘shape‘

测试程序有一句出现了bug，以前没见过，就是正常使用，不觉得会有错。

import torch

b = torch.range(1,12,1).reshape([3,2,2])
print(b,b.shape)

d = torch.max(b,dim=0,keepdim=True)###bug所在

print(d,d.shape)

网上查了一下，torch.max（）返回是个turtle，包含了 value，index，所以错误原因也找到了，没有定义index的变量。修改后如下：

import torch

b = torch.range(1,12,1).reshape([3,2,2])
print(b,b.shape)

d,e = torch.max(b,dim=0,keepdim=True)

print(d,d.shape)
print(e)
#######################
E:\Anconda\python.exe "C:/Users/MR-LI/Desktop/program practice/1.py"
tensor([[[ 1.,  2.],
C:/Users/MR-LI/Desktop/program practice/1.py:3: UserWarning: torch.range is deprecated in favor of torch.arange and will be removed in 0.5. Note that arange generates values in [start; end), not [start; end].
         [ 3.,  4.]],
  b = torch.range(1,12,1).reshape([3,2,2])

        [[ 5.,  6.],
         [ 7.,  8.]],

        [[ 9., 10.],
         [11., 12.]]]) torch.Size([3, 2, 2])
tensor([[[ 9., 10.],
         [11., 12.]]]) torch.Size([1, 2, 2])
tensor([[[2, 2],
         [2, 2]]])

OK了！！！！