结论:
torch.max(),不指定dim参数,返回输入张量中所有数据的最大值; 如果指定dim参数,则返回按照指定维度的最大值和各最大值对应的位置
torch.argmax(),不指定dim,然后输入张量中所有数据最大值位置(按照张量被拉伸为1为向量算); 如果指定dim,返回指定维度的最大值位置,另外可以通过keepdim来保留原张量的形状
2023-12-26 20:46:57
结论:
torch.max(),不指定dim参数,返回输入张量中所有数据的最大值; 如果指定dim参数,则返回按照指定维度的最大值和各最大值对应的位置
torch.argmax(),不指定dim,然后输入张量中所有数据最大值位置(按照张量被拉伸为1为向量算); 如果指定dim,返回指定维度的最大值位置,另外可以通过keepdim来保留原张量的形状