建立网络,对pytorch学习
torch.max情况1:返回最大值情况2:返回torch.return_types.max,这里有两个维度,第一个维度是该向量的最大值,第二个维度是该向量最大值所在索引。假设有mxn的矩阵,则得到m个最大值和对应索引。==这样的好处,既可以找到最大值,又可以找到最大值所在索引。例子:outputs = net(im) # [1,10]predict = torch.max(outputs, dim=1) # 返回torch.return_types.max,这里有两个维度,第一个维度是该向量的.
原创
2020-12-28 19:20:06 ·
964 阅读 ·
0 评论