代码之家 › 专栏 › 技术社区 › Nico Schlömer David Maze

一次计算所有外积[重复]

numpy arrays python

Nico Schlömer David Maze · 技术社区 · 7 年前

a 形状 (n, k1) b 形状 (n, k2) ,我想计算 a[i] 和 b[i]

import numpy

n = 3

a = numpy.random.rand(n, 7)
b = numpy.random.rand(n, 11)

out = numpy.array([
    numpy.outer(a[i], b[i]) for i in range(n)
    ])

print(out.shape)  # (n, 7, 11)

这是一个技巧,但它包含一个Python循环,大大降低了速度 n .

外积可以立即计算吗?

1 回复 | 直到 7 年前

cs95 7 年前

np.einsum('ij,ik->ijk', a, b)

或者,使用广播,

a[:,:,np.newaxis] * b[:,np.newaxis,:]

推荐文章

July · 如何定义数字间隔,然后四舍五入

1 年前

Community wiki · 对象名称前的单下划线和双下划线的含义是什么?

1 年前

Brian Johnson · 为什么在Python中列出字典列表会引发TypeError?[已关闭]

1 年前

user026 · 如何根据特定窗口的平均值(行数)创建新列?

1 年前

Ashok Shrestha · 需要追踪特定的颜色线并获取坐标

1 年前

Nicote Ool · 在FastApi和Vue3中获得422

1 年前

NeoExceptCodeBad · 如果我有很多垂直线,我如何找到它们的边缘?

1 年前

Abdulaziz · 如何对集合内的列表进行排序[重复]

1 年前

user2743931 · 带有src目录的Python setup.py

1 年前

asmgx · 为什么合并数据帧不能按照python中的预期方式工作

1 年前