1

テンソルに行列を追加したいと思います。

これは動作します

numpy.zeros((4,4,4))+numpy.sum(numpy.ones((4,4,4)),axis=2)

これは機能しません。

numpy.zeros((3,4,5))+numpy.sum(numpy.ones((3,4,5)),axis=2)
ValueError: operands could not be broadcast together with shapes (3,4,5) (3,4)

2番目の例を機能させる方法は?3番目の軸に沿って追加したいと思います。

4

1 に答える 1

4

ブロードキャスト

numpy.zeros((3,4,5))+numpy.sum(numpy.ones((3,4,5)),axis=2)[...,None]
于 2013-03-25T13:10:19.673 に答える