f**********d
发帖数: 4960 | 1 a是k维矢量
b是k*i*j维array
想用a给b的第二第三维组成的矩阵加权
即用a1给b【1，：，：】加权，and so on。
这个用哪个numpy函数可以方便地得到？ | d*****u
发帖数: 17243 | 2 aaa = numpy.expand_dims(numpy.expand_dims(a, axis=1), axis=2)
result = aaa * b | f**********d
发帖数: 4960 | 3 Thx
查了一下a[:,np.newaxis, np.newaxis]即可
和你写的一样
【在 d*****u 的大作中提到】
: aaa = numpy.expand_dims(numpy.expand_dims(a, axis=1), axis=2)
: result = aaa * b
|
|
