NumPy的np.dot函数:计算点积与矩阵乘积的利器

np.dot 是 NumPy 库中的一个函数,用于计算两个数组的点积。它支持多种类型的输入数组,包括向量和矩阵,并返回它们的点积或矩阵乘积。

对于两个向量(一维数组),np.dot 返回它们的点积,即对应元素的乘积之和。对于矩阵和向量的乘积,或者两个矩阵的乘积,np.dot 执行标准的线性代数矩阵乘法。

下面是一些 np.dot 的基本用法示例:

两个向量的点积

复制代码

python复制代码

|---|----------------------------|
| | import numpy as np |
| | |
| | a = np.array([1, 2, 3]) |
| | b = np.array([4, 5, 6]) |
| | |
| | result = np.dot(a, b) |
| | print(result) # 输出:32 |

矩阵与向量的乘积

复制代码

python复制代码

|---|-----------------------------------|
| | A = np.array([[1, 2], [3, 4]]) |
| | b = np.array([5, 6]) |
| | |
| | result = np.dot(A, b) |
| | print(result) # 输出:[17 39] |

两个矩阵的乘积

复制代码

python复制代码

|---|-----------------------------------|
| | A = np.array([[1, 2], [3, 4]]) |
| | B = np.array([[5, 6], [7, 8]]) |
| | |
| | result = np.dot(A, B) |
| | print(result) |
| | # 输出: |
| | # [[19 22] |
| | # [43 50]] |

请注意,当计算两个矩阵的乘积时,左侧矩阵的列数必须等于右侧矩阵的行数,否则 NumPy 会抛出一个错误。

np.dot 在进行数组运算时非常高效,因为它利用了底层的优化算法。在处理大型数组或进行复杂的数学运算时,使用 np.dot 通常比手动实现这些运算要快得多。

相关推荐
dllmayday9 小时前
3D空间中的变换矩阵
线性代数·3d·矩阵
是数学系的小孩儿13 小时前
论文:M矩阵
线性代数·矩阵
冲帕Chompa13 小时前
今日矩阵系列
数据结构·算法·矩阵
xueyongfu15 小时前
PTX指令集基础以及warp级矩阵乘累加指令介绍
人工智能·线性代数·算法·矩阵
小妖同学学AI21 小时前
deepseek+飞书多维表格 打造小红书矩阵
人工智能·矩阵·飞书
细嗅蔷薇@1 天前
C语言在键盘上输入一个3行3列矩阵的各个元素的值(值为整数),然后输出主对角线元素的积,并在fun()函数中输出。
c语言·算法·矩阵
monicaaaaan1 天前
搜索二维矩阵Ⅱ C++
c++·线性代数·矩阵
闲看云起1 天前
从矩阵表示到卷积神经网络(CNN)与循环神经网络(RNN)
人工智能·rnn·矩阵·cnn
go54631584651 天前
离散扩散模型在数独问题上的复现与应用
线性代数·算法·yolo·生成对抗网络·矩阵
设计师小聂!1 天前
力扣热题100-------74.搜索二维矩阵
算法·leetcode·矩阵