torch.max() x.item() x.sum() x.size(0) x.squeeze()

torch.max()

torch.max(a,0)或torch.max(a,1)
torch.max(a,0)返回行的最大值+最大值的索引
torch.max(a,1)返回列的最大值+最大值的索引

参考博客

x.item()

x.item() →获取x的值 只能是数值，不能是list
x必须是tensor 且只能是数值

x.sum()

x.sum() →计算x的总和
x是tensor
可以x.sum().item()连用，先求和再获取值

x.size(0)

x.size()返回张量tensor的shape
x.size(0)返回shape的第0维度
x.size(-1)返回shape的最后一个维度

import torch

yyy = torch.tensor(111)
print(yyy.size())
# print(yyy.size(0))  # 报错 因为没有shape
# print(yyy.size(-1)) # 报错