pycharm报错ValueError: shape mismatch: objects cannot be broadcast to a single shape. Mismatch is between arg 0 with shape (31, 41) and arg 1 with shape (31, 41, 85).
时间: 2023-07-09 14:28:23 浏览: 380
这个报错的意思是你正在尝试将两个不同形状的张量进行广播,但是这两个张量的形状不兼容。具体来说,arg 0 的形状是 (31, 41),而 arg 1 的形状是 (31, 41, 85)。
要解决这个问题,你需要确保这两个张量的形状相同,或者通过 numpy 的 broadcasting 功能将它们转换成相同形状。你可以使用 np.expand_dims() 函数增加 arg 0 的维度,使其形状变为 (31, 41, 1),从而与 arg 1 的形状兼容。具体实现代码如下:
```python
import numpy as np
arg0 = np.random.rand(31, 41)
arg1 = np.random.rand(31, 41, 85)
# 将 arg0 的形状变为 (31, 41, 1)
arg0 = np.expand_dims(arg0, axis=-1)
# 现在 arg0 和 arg1 的形状相同,可以进行广播操作了
result = arg0 * arg1
```
阅读全文