【torch.max】TypeError: expected Tensor as element 1 in argument 0, but got torch.return_types.max

最新推荐文章于 2024-09-27 20:31:54 发布

qian9905

最新推荐文章于 2024-09-27 20:31:54 发布

阅读量3.2k

点赞数 4

分类专栏：深度学习填坑记录文章标签： python 深度学习 pytorch

本文链接：https://blog.youkuaiyun.com/qian9905/article/details/128376355

版权

深度学习填坑记录专栏收录该内容

4 篇文章

订阅专栏

这篇博客主要探讨了在使用PyTorch时遇到的一个常见误区，即对torch.max函数返回值的理解不清晰。torch.max()不仅返回最大值，还会返回最大值的索引。通过修正代码，将torch.max().values用于获取最大值tensor，避免了错误。文章强调了正确理解和使用这个函数对于计算操作的重要性，特别是在计算均值和最大值的结合过程中。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

实际上这是一个max返回值认知不清的误区
我遇到该报错原因是，torch.max函数返回值包含两个：第一个是max的tensor,第二个是max值的位置index.返回的最大值和索引各是一个tensor，分别表示该维度的最大值，以及该维度最大值的索引，一起构成元组(Tensor, LongTensor)
torch.max().values,可以获取最大值tensor

torch.max().indices可以获取最大值索引tensor

报错代码及修改如下

        fmean = torch.mean(feats, dim=1, keepdim=True)
        #原代码：fmax = torch.max(feats, dim=1, keepdim=True)
        fmax = torch.max(feats, dim=1, keepdim=True).values#修改后
        fsum = torch.cat((fmean, fmax), dim=1)#报错行