AttributeError: 'Softmax' object has no attribute 'size'
时间: 2023-06-21 13:07:31 浏览: 635
python报错: 'list' object has no attribute 'shape'的解决
5星 · 资源好评率100%
这个错误通常是因为你试图使用 PyTorch 中的 `size()` 方法,但是你的对象是一个 Softmax 层,并不支持 `size()` 方法。
要获取 Softmax 层的输出形状,你可以使用 `output_shape` 属性来获取。比如:
```python
import torch.nn as nn
softmax_layer = nn.Softmax(dim=1)
x = torch.randn(1, 10)
y = softmax_layer(x)
print(y)
print(y.shape) # 打印输出张量的形状
```
在上面的代码中,我们首先创建了一个 Softmax 层,并将其应用于输入张量 `x` 上。然后,我们打印了输出张量 `y` 的形状,以确认它的大小。
阅读全文