学习笔记(25):线性代数,矩阵-矩阵乘法原理

学习笔记(25):线性代数,矩阵-矩阵乘法原理

1、代码

python 复制代码
import torch
A = torch.arange(20).reshape(5, 4)
print(A)
A = A.to(torch.float32)  # 或使用A.float()

B = torch.ones(4, 3)
print(B)
print(torch.mm(A, B))

2、执行结果

tensor([[ 0, 1, 2, 3],

4, 5, 6, 7\], \[ 8, 9, 10, 11\], \[12, 13, 14, 15\], \[16, 17, 18, 19\]\]) tensor(\[\[1., 1., 1.\], \[1., 1., 1.\], \[1., 1., 1.\], \[1., 1., 1.\]\]) tensor(\[\[ 6., 6., 6.\], \[22., 22., 22.\], \[38., 38., 38.\], \[54., 54., 54.\], \[70., 70., 70.\]\])

计算原理解释:

要计算 torch.mm(A, B)(矩阵乘法),需要确保 A 的列数等于 B 的行数。在你的例子中:

  • A 是一个 5×4 的矩阵(5 行 4 列)。
  • B 是一个 4×3 的矩阵(4 行 3 列)。

由于 A 的列数(4)等于 B 的行数(4) ,因此可以进行矩阵乘法,结果将是一个 5×3 的矩阵。

矩阵乘法原理

矩阵乘法 C = A × B 的计算规则是:

  • C 的行数等于 A 的行数(5 行)。
  • C 的列数等于 B 的列数(3 列)。
  • C 中每个元素 C[i,j] 等于 A 的第 i 行与 B 的第 j 列对应元素的乘积之和

计算步骤示例

以结果矩阵 C 的第一行第一列元素 C[0,0] 为例:

  1. 取 A 的第 0 行[0, 1, 2, 3]
  2. 取 B 的第 0 列[1., 1., 1., 1.]
  3. 对应元素相乘后求和:\((0 \times 1) + (1 \times 1) + (2 \times 1) + (3 \times 1) = 0 + 1 + 2 + 3 = 6\)

同理,计算其他元素:

  • C[0,1] = (0×1) + (1×1) + (2×1) + (3×1) = 6
  • C[0,2] = (0×1) + (1×1) + (2×1) + (3×1) = 6
  • 以此类推,最终得到整个矩阵 C

A=tensor([[ 0, 1, 2, 3],

4, 5, 6, 7\], \[ 8, 9, 10, 11\], \[12, 13, 14, 15\], \[16, 17, 18, 19\]\]) B=tensor(\[\[1., 1., 1.\], \[1., 1., 1.\], \[1., 1., 1.\], \[1., 1., 1.\]\]) 结果: tensor(\[\[ 6., 6., 6.\], \[22., 22., 22.\], \[38., 38., 38.\], \[54., 54., 54.\], \[70., 70., 70.\]\])

验证方法

由于 B 的所有元素都是 1,因此结果矩阵 C 的每个元素 C[i,j] 实际上等于 A 的第 i 行的所有元素之和。例如:

  • A 的第 0 行和为 0+1+2+3=6,对应 C [0,:] 的所有元素。
  • A 的第 1 行和为 4+5+6+7=22,对应 C [1,:] 的所有元素。
  • 以此类推。
相关推荐
Asu520225 分钟前
思途SQL学习 0729
数据库·sql·学习
Star在努力27 分钟前
15-C语言:第15~16天笔记
c语言·笔记·算法
go54631584651 小时前
大规模矩阵构建与高级算法应用
线性代数·算法·矩阵
ZY小袁2 小时前
MGRE综合实验
服务器·网络·笔记·网络安全·学习方法·信息与通信·p2p
一位搞嵌入式的 genius2 小时前
暑期自学嵌入式——Day10(C语言阶段)
linux·笔记·学习·嵌入式c语言
被遗忘的旋律.2 小时前
Linux驱动开发笔记(五)——设备树(上)
linux·驱动开发·笔记
霜绛3 小时前
机器学习笔记(四)——聚类算法KNN、Kmeans、Dbscan
笔记·算法·机器学习·kmeans·聚类
晨非辰4 小时前
#C语言——学习攻略:深挖指针路线(三)--数组与指针的结合、冒泡排序
c语言·开发语言·数据结构·学习·算法·排序算法·visual studio
先生沉默先4 小时前
Docker学习日志-Docker容器配置、Nginx 配置与文件映射
学习·nginx·docker
Brookty10 小时前
Java线程安全与中断机制详解
java·开发语言·后端·学习·java-ee