如何在numpy中沿轴添加

时间:2013-03-25 13:07:54

标签: python numpy

我想在张量中添加一个矩阵。

这有效

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)

如何让第二个例子正常工作?我想沿第三轴添加。

1 个答案:

答案 0 :(得分:3)

Broadcast

numpy.zeros((3,4,5))+numpy.sum(numpy.ones((3,4,5)),axis=2)[...,None]