You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

如何用向量化方法计算两个D×N矩阵列向量的叉积?

向量化实现两个3×N矩阵列向量的叉积

没问题!要高效实现两个D×N矩阵列向量的叉积,首先得明确一个前提:叉积仅在3维空间(即D=3)中存在定义,所以咱们默认你的两个输入矩阵都是3×N的形状(每一列是一个3维向量)。下面直接给你用NumPy实现的向量化方案,完全不用写循环,效率拉满:

  • 首先导入NumPy库:
import numpy as np
  • 构造示例的3×N矩阵(你可以替换成自己的实际数据):
# 生成两个3行5列的随机矩阵,每列对应一个3维向量
A = np.random.rand(3, 5)
B = np.random.rand(3, 5)
  • 核心的向量化叉积计算:
# 指定axis=0,让NumPy沿着行方向(也就是对每一列的3个元素)计算叉积
cross_product = np.cross(A, B, axis=0)

关键说明

这里的axis=0是重点:它告诉NumPy以列为单位进行叉积运算,最终得到的cross_product同样是3×N的矩阵,每一列就是A和B对应列向量的叉积结果。

如果你的矩阵是N×3的(行向量形式),只需要把axis参数改成1就可以啦。另外要是你的D不是3,叉积本身是没有数学定义的,这时候得先确认是不是在3维空间下的运算需求哦~

内容的提问来源于stack exchange,提问作者Darkmoor

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.05.19 07:33:54