积在Numpy的第二维上吗?

问题描述 投票:0回答:1

我有N个向量的数组,每个向量的大小为3:]

[[ 0  1  2]
 [ 3  4  5]
 [ 6  7  8]
 [ 9 10 11]
 [12 13 14]]

需要找到每个向量与其自身的点积。因此,结果将是形状(N,3,3)的3维数组。一种方法是使用以下for循环:

for vector in np.arange(15).reshape(-1,3):
    np.outer(vector, vector)

由于数组可以任意大,所以我需要找到一个向量化的解决方案。

我有N个向量的数组,每个向量的大小为3:[[0 1 2] [3 4 5] [6 7 8] [9 10 11] [12 13 14]]需要找到一个每个向量与其自身的点积。因此,...

python numpy
1个回答
2
投票

您没有对任何轴求和,对吗?只是一个“划痕的”外部?

© www.soinside.com 2019 - 2024. All rights reserved.