torch.max()的使用讲解
函数一般使用方法:output, index = torch.max(input, dim)解释:torch.max()输入:input和diminput:是一个需要进行数据处理的tensordim:等于0时为求每列的最大值,等于1时为求每行的最大值(容易弄混一定要记牢!)torch.max()函数返回的有两个tensor,第一个是每行或每列的最大值,第二个是该最大值在该行或该列的索引。...
原创
2021-06-27 22:48:08 ·
583 阅读 ·
0 评论