AttributeError: ‘torch.return_types.max‘ object has no attribute ‘dim‘

我对算法一无所知

于 2021-01-22 10:30:28 发布

阅读量5.3k

点赞数 5

分类专栏：各种error总结文章标签： pytorch

本文链接：https://blog.youkuaiyun.com/qq_31267769/article/details/112971077

版权

在使用PyTorch时遇到AttributeError，问题出在尝试访问'torch.return_types.max'对象的'dim'属性，但该对象并不包含此属性。官方文档说明torch.max返回的是一个元组，包含最大值和最大值的索引。若只需要获取最大值，应当正确处理返回的元组。

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

out1 = torch.max(out1, 1) # (batch, embedding_size)
out1 = self.linear(out1) # (batch, linear_hidden_size)

抛出错误

  File "/diske/Anaconda3/envs/pytorch1.4/lib/python3.7/site-packages/torch/nn/modules/module.py", line 722, in _call_impl
    result = self.forward(*input, **kwargs)
  File "lstm.py", line 75, in forward
    out1 = self.linear(out1) # (batch, linear_hidden_size)
  File "/diske/Anaconda3/envs/pytorch1.4/lib/python3.7/site-packages/torch/nn/modules/module.py", line 722, in _call_impl
    result = self.forward(*input, **kwargs)
  File "/diske/Anaconda3/envs/pytorch1.4/lib/python3.7/site-packages/torch/nn/modules/linear.py", line 91, in forward
    return F.linear(input, self.weight, self.bias)
  File "/diske/Anaconda3/envs/pytorch1.4/lib/python3.7/site-packages/torch/nn/functional.py", line 1672, in linear
    if input.dim() =

最低0.47元/天解锁文章