在Numpy阵列广播中考虑以下练习.
import numpy as np v = np.array([[1.0,2.0]]).T # column array A2 = np.random.randn(2,10) # 2D array A3 = np.random.randn(2,10,10) # 3D v * A2 # works great # causes error: v * A3 # error
我知道广播的Numpy规则,我熟悉Matlab中的bsxfun功能.我明白为什么试图将(2,1)数组广播到(2,N,N)数组中失败,我必须在(2,1)数组之前将(2,1)数组重新形成广播经过.
我的问题是:有没有办法告诉Python在尝试广播时自动填充数组的维度,而我不必专门告诉它必要的维度?
我不想明确地将(2,1)向量与要广播的多维数组对齐 – 否则我可以做一些像mult_v_A = lambda v,A:v.reshape([ v.size] [1] *(A.ndim-1))* A.如果“A”数组将是2D或3D或ND,我不知道如何.
Matlab的bsxfun广播功能根据需要隐含地填充维度,所以我希望有一些我可以在Python中做的事情.
解决方法
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。