a=torch.randn(1,2,3)
b=torch.max(a,dim=0).values
c=torch.max(a,dim=1).values
d=torch.max(a,dim=2).values
b.shape=[2,3]
c.shape=[1,3]
d.shape=[1,2]
a=torch.randn(1,2,3)
b=torch.max(a,dim=0).values
c=torch.max(a,dim=1).values
d=torch.max(a,dim=2).values
b.shape=[2,3]
c.shape=[1,3]
d.shape=[1,2]