矩阵相乘torch.einsum()

einsumEinstein summation 的缩写,来源于爱因斯坦求和约定(Einstein summation convention)。这是物理学家阿尔伯特·爱因斯坦引入的一种简便记号,用于描述张量运算,特别是涉及多维数组的运算。

示例1:矩阵乘法

矩阵乘法 C=AB

A = torch.randn(2, 3)
B = torch.randn(3, 4)
C = torch.einsum('ik,kj->ij', A, B)
print(C.size())  # 输出: torch.Size([2, 4])

 这里,'ik,kj->ij' 的含义是:

  • A 的形状为 (2, 3),对应 ikik 分别表示第一个和第二个维度。
  • B 的形状为 (3, 4),对应 kjkj 分别表示第一个和第二个维度。
  • ->ij 表示输出张量的模式,结果为 (2, 4)
示例2:向量点积

向量点积 c=a⋅b

a = torch.randn(3)
b = torch.randn(3)
c = torch.einsum('i,i->', a, b)
print(c.size())  # 输出: torch.Size([])

这里,'i,i->' 的含义是:

  • ab 都是向量,对应模式 i
  • -> 后面为空,表示结果是一个标量。
示例3:批量矩阵乘法

批量矩阵乘法

A = torch.randn(10, 2, 3)
B = torch.randn(10, 3, 4)
C = torch.einsum('bij,bjk->bik', A, B)
print(C.size())  # 输出: torch.Size([10, 2, 4])

这里,'bij,bjk->bik' 的含义是:

  • A 的形状为 (10, 2, 3),对应 bijb 表示批次维度,ij 分别表示矩阵的行和列。
  • B 的形状为 (10, 3, 4),对应 bjkb 表示批次维度,jk 分别表示矩阵的行和列。
  • ->bik 表示输出张量的模式,结果为 (10, 2, 4)

示例4:逐元素相乘(哈达玛积)A.B或A × B

A = torch.randn(3, 4)
B = torch.randn(3, 4)

C = torch.einsum('ij,ij->ij', A, B)
print(C.size())  # 输出: torch.Size([3, 4])

'ij,ij->ij' 表示:

  • AB 都是形状为 [3, 4] 的矩阵,用 ij 表示。
  • 结果 C 也是形状为 [3, 4] 的矩阵。
  • 没有重复索引,所以不进行求和。

 

相关推荐

  1. python矩阵相乘

    2024-06-08 18:50:03       40 阅读
  2. 矩阵相乘torch.einsum()

    2024-06-08 18:50:03       47 阅读
  3. 题目 1908: 蓝桥杯-矩阵相乘

    2024-06-08 18:50:03       40 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-06-08 18:50:03       169 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-06-08 18:50:03       185 阅读
  3. 在Django里面运行非项目文件

    2024-06-08 18:50:03       155 阅读
  4. Python语言-面向对象

    2024-06-08 18:50:03       169 阅读

热门阅读

  1. mybatisplus QueryWrapper or 写法

    2024-06-08 18:50:03       44 阅读
  2. window.clearInterval(timer) 清除定时器

    2024-06-08 18:50:03       40 阅读
  3. Docker

    Docker

    2024-06-08 18:50:03      40 阅读
  4. Redis命令使用示例(一)

    2024-06-08 18:50:03       38 阅读
  5. Ansible——user模块

    2024-06-08 18:50:03       44 阅读
  6. SD-WAN加速跨国服务器访问

    2024-06-08 18:50:03       36 阅读
  7. Spring Boot中实现规则引擎源码教程

    2024-06-08 18:50:03       35 阅读
  8. android:text 总为大写字母的原因

    2024-06-08 18:50:03       36 阅读
  9. input只允许输入数字

    2024-06-08 18:50:03       35 阅读
  10. 数据结构汇总学习(ing)

    2024-06-08 18:50:03       31 阅读
  11. 注解 - @RequestBody

    2024-06-08 18:50:03       41 阅读