torch.max和torch.argmax,结论:torch.max(),不指定dim参数,返回输入张量中所有数据的最大值;如果指定dim参数,则返回按照指定维度的最大值和各 ...
確定! 回上一頁