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

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_implresult = self.forward(*input, **kwargs)File "lstm.py", line 75, in forwardout1 = 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_implresult = self.forward(*input, **kwargs)File "/diske/Anaconda3/envs/pytorch1.4/lib/python3.7/site-packages/torch/nn/modules/linear.py", line 91, in forwardreturn 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 linearif input.dim() == 2 and bias is not None:
AttributeError: 'torch.return_types.max' object has no attribute 'dim'

发现官网对torch.max()的解释是这样：

torch.max 返回值是一个tuple，第一个元素是max值，第二个元素是max值的索引。

>>> a = torch.randn(4, 4)
>>> a
tensor([[-1.2360, -0.2942, -0.1222,  0.8475],[ 1.1949, -1.1127, -2.2379, -0.6702],[ 1.5717, -0.9207,  0.1297, -1.8768],[-0.6172,  1.0036, -0.6060, -0.2432]])
>>> torch.max(a, 1)
torch.return_types.max(values=tensor([0.8475, 1.1949, 1.5717, 1.0036]), indices=tensor([3, 0, 0, 1]))

只需要获得最大值即可

out1 = torch.max(out1, 1) # (batch, embedding_size)
out1 = self.linear(out1.values) # (batch, linear_hidden_size)
# 或者
val, ind = torch.max(out1, 1) # (batch, embedding_size)
out1 = self.linear(val) # (batch, linear_hidden_size)

AttributeError: ‘torch.return_types.max‘ object has no attribute ‘dim‘相关推荐

AttributeError: 'PyQt5.QtCore.pyqtSignal' object has no attribute 'connect'
AttributeError: 'PyQt5.QtCore.pyqtSignal' object has no attribute 'connect' https://blog.csdn.net/je ...
M1Mac使用TensorFlow出现AttributeError: ‘h5py.h5t.TypeID‘ object has no attribute ‘set_fields‘的解决办法
问题 TensorFlow使用Apple官方优化版2.4.1(conda-forge),使用tf.keras后端时候出现: AttributeError: 'h5py.h5t.TypeID' obje ...
AttributeError: ‘tensorrt.tensorrt.Builder‘ object has no attribute ‘max_workspace_size‘
TensorRT版本:8.4 从网上下载的代码,运行报错,如上述标题. 1.错误:AttributeError: 'tensorrt.tensorrt.Builder' object has no a ...
AttributeError: ‘list’ object has no attribute “dim”
这错是因为居然自己蠢到在pytorch用list当tensor用将get_image转成tensor即可 get_image=torch.from_numpy(numpy.array(get_ima ...
问题：AttributeError: 'Tensor' object has no attribute 'creator'
AttributeError: 'Tensor' object has no attribute 'creator' 根据pytorch官方文档的说法,变量具有如上的三个属性,在获取y操作的creat ...
AttributeError: ‘str‘ object has no attribute ‘copy
在加载模型时出现 AttributeError: 'str' object has no attribute 'copy' 错误解决方案: 将 model.load_state_dict(state ...
pytorch版本问题：AttributeError: 'module' object has no attribute '_rebuild_tensor_v2'
用pytorch加载训练好的模型的时候遇到了如下的问题: AttributeError: 'module' object has no attribute '_rebuild_tensor_v2' 到 ...
AttributeError: ‘Embeddings‘ object has no attribute ‘d_model‘
求助AttributeError: 'Embeddings' object has no attribute 'd_model' embedding.d_model不是transformer_util ...
使用torchsummary时出现AttributeError: ‘list‘ object has no attribute ‘size‘解决方案
目录 1 问题 2 原因 3 解决方案 4 代码 1 问题使用torchsummary查看网络结构时报错:AttributeError: 'list' object has no attribute ...
Jetson Nano【13】关于torch2trt报错：AttributeError: ‘Tensor‘ object has no attribute ‘_trt‘的一种可能性
尝试转换,报错,日志如下 AttributeError Traceback (most recent call last) <ipython-input-24-0cf9bfd4fe52> ...

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

AttributeError: ‘torch.return_types.max‘ object has no attribute ‘dim‘相关推荐

最新文章

热门文章