【python深度学习】——torch.einsum|torch.bmm

【python深度学习】——torch.einsum|torch.bmm

  • 1. 基本用法与示例
  • 2. torch.bmm

1. 基本用法与示例

基本用法:

torch.einsum(equation, *operands)
  • equation: 一个字符串,定义了张量操作的模式。
    使用逗号来分隔输入张量的索引,然后是一个箭头(->),接着是输出张量的索引
  • operands: 要操作的张量。
    示例代码:
import torch
A = torch.randn(2, 3)B = torch.einsum('ij->ji', A)
# 等价于 B = A.transpose(0, 1)C = torch.einsum('ik,kj->ij', A, B)
# 等价于 C = torch.matmul(A, B)a = torch.randn(3)
b = torch.randn(3)
c = torch.einsum('i,i->', a, b)
# 等价于 c = torch.dot(a, b)A = torch.randn(5, 2, 3)
B = torch.randn(5, 3, 4)
C = torch.einsum('bij,bjk->bik', A, B)
# 等价于 C = torch.bmm(A, B)a = torch.randn(3)
b = torch.randn(4)
c = torch.einsum('i,j->ij', a, b)
# 结果是一个3x4的矩阵,等价于 c = a.unsqueeze(1) * b.unsqueeze(0)A = torch.randn(3, 3)
trace = torch.einsum('ii->', A)
# 等价于 trace = torch.trace(A)

2. torch.bmm

全称为: batch matrix-matrix product, 批量矩阵乘法, 适用于三维张量,其中第一维表示批量大小,第二维和第三维表示矩阵的行和列

torch.bmm(input, mat2, *, out=None) -> Tensor
  • input: 一个形状为 (b, n, m) 的三维张量,表示一批矩阵。
  • mat2: 一个形状为 (b, m, p) 的三维张量,表示另一批矩阵。
  • out (可选): 存储输出结果的张量。
    输出是一个形状为 (b, n, p) 的张量,其中每个矩阵是对应批次的矩阵乘法结果。

例如:

import torch# 定义两个形状为 (b, n, m) 和 (b, m, p) 的三维张量
batch_size = 10
n, m, p = 3, 4, 5A = torch.randn(batch_size, n, m)
B = torch.randn(batch_size, m, p)# 进行批量矩阵乘法
C = torch.bmm(A, B)print(C.shape)  # 输出: torch.Size([10, 3, 5])

再具体的:

A = torch.tensor([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
B = torch.tensor([[[9, 10], [11, 12]], [[13, 14], [15, 16]]])# A.shape = (2, 2, 2)
# B.shape = (2, 2, 2)
C = torch.bmm(A, B)print(C)
# 输出:
# tensor([[[ 31,  34],
#          [ 73,  80]],
#
#         [[155, 166],
#          [211, 226]]])

其数学计算为:
请添加图片描述

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/pingmian/25236.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

linux中dd命令以及如何测试读写速度

dd命令详解 dd命令是一个在Unix和类Unix系统中非常常用的命令行工具,它主要用于复制文件和转换文件数据。下面我会详细介绍一些dd命令的常见用法和功能: 基本语法 dd命令的基本语法如下: bash Copy Code dd [option]...主要选项和参数 if…

Data Mining2 复习笔记6 - Optimization Hyperparameter Tuning

6. Optimization & Hyperparameter Tuning Why Hyperparameter Tuning? Many learning algorithms for classification, regression, … Many of those have hyperparameters: k and distance function for k nearest neighbors, splitting and pruning options in decis…

力扣1712.将数组分成三个子数组的方案数

力扣1712.将数组分成三个子数组的方案数 确定左边界的值 然后二分求右边界的范围 右边界处的前缀和满足 2*s[i] < s[r] < (s[n] s[i]) / 2 int s[100010];const int N 1e97;class Solution {public:int waysToSplit(vector<int>& nums) {int n nums.siz…

Mac清洁神器CleanMyMac2024一键轻松解决电脑垃圾问题

【CleanMyMac】苹果mac电脑垃圾清理软件 大家好&#xff01;今天我要给大家种草一个超级好用的苹果mac电脑垃圾清理软件&#xff0c;那就是 CleanMyMac。相信很多人都遇到过电脑运行速度变慢、存储空间不足的问题&#xff0c;而这款软件就是解决这些问题的救星&#xff01;让我…

2024眼睛健康展|江苏省护眼产品与眼部保健展会|眼科医疗展

呵护眼睛、守护光明&#xff0c;2024全国眼睛健康产业博览会暨眼科医学大会10月11日在南京启幕&#xff1b; 办展&#xff1a;随着时代的进步、社会的发展&#xff0c;特别是电子产品的深度普及&#xff1b;近些年&#xff0c;人们的用眼时间越来越久&#xff0c;由此产生高发…

一个python 程序执行顺序

1. Python程序执行顺序 在Python中&#xff0c;程序的执行顺序通常遵循几个基本原则&#xff1a; &#xff08;1&#xff09;从上到下&#xff1a;Python代码通常从上到下顺序执行。 &#xff08;2&#xff09;代码块&#xff1a;由缩进&#xff08;如空格或制表符&#xff…

20240605解决飞凌的OK3588-C的核心板刷机原厂buildroot不能连接ADB的问题

20240605解决飞凌的OK3588-C的核心板刷机原厂buildroot不能连接ADB的问题 2024/6/5 13:53 rootrootrootroot-ThinkBook-16-G5-IRH:~/repo_RK3588_Buildroot20240508$ ./build.sh --help rootrootrootroot-ThinkBook-16-G5-IRH:~/repo_RK3588_Buildroot20240508$ ./build.sh lun…

基于I2C协议的OLED显示(利用U82G库)

目录 一、I2C协议的基本原理和时序协议I2C通信协议的原理I2C时序基本单元I2C时序 二、建立工程RCC配置TIM1配置时钟树配置工程配置 三、U8g2移植精简u8g2_d_setup.c精简u8g2_d_memory.c编写移植函数stm32_u8g2.hstm32_u8g2.c 四、实验1.U82G的demo例程2.显示网名昵称中文取模步…

L48---1637. 两点之间不包含任何点的最宽垂直区域(排序)---Java版

1.题目描述 2.思路 &#xff08;1&#xff09;返回两点之间内部不包含任何点的 最宽垂直区域 的宽度。 我的理解是相邻两个点&#xff0c;按照等差数列那样&#xff0c;后一个数减去相邻的前一个数&#xff0c;才能保证两数之间不含其他数字。 &#xff08;2&#xff09;所以&…

c++|unordered系列关联式容器(unordered_set、unordered_map介绍使用+哈希结构)

目录 一、unordered_set的介绍与使用 1.1unordered_set介绍 1.2unordered_set使用 2.2.1构造 2.2.2容量 2.2.3修改 二、unordered_map的介绍与使用 2.1unordered_map介绍 2.2unordered_map使用 2.2.1构造 2.2.2容量 2.2.3修改 三、底层结构(哈希) 3.1哈希概念 3.2哈…

【回调函数】

1.回调函数是什么&#xff1f; 回调函数就是⼀个通过函数指针调用的函数。 如果你把函数的指针&#xff08;地址&#xff09;作为参数传递给另⼀个函数&#xff0c;当这个指针被用来调用其所指向的函数 时&#xff0c;被调用的函数就是回调函数。回调函数不是由该函数的实现方…

【PL理论】(12) F#:模块 | 命名空间 | 异常处理 | 内置异常 |:? | 相互递归函数

&#x1f4ad; 写在前面&#xff1a;本章我们将介绍 F# 的模块&#xff0c;我们前几章讲的列表、集合和映射都是模块。然后我们将介绍 F# 中的异常&#xff0c;以及内置异常&#xff0c;最后再讲解一下相互递归函数。 目录 0x00 F# 模块&#xff08;Module&#xff09; 0x01…

Bootstrap框架集成ECharts教程

最新公司项目要在原有的基础上增加一些饼状图和柱状图来统计一些数据给客户&#xff0c;下面就是集成的一个过程&#xff0c;还是很简单的&#xff0c;分为以下几步 1、引入ECharts的包 2、通过ECharts官网或者菜鸟教程直接拿示例代码过来修修改改直接用就可以了 注意&#xf…

Windows关闭自动更新最有效的方法

按WR打开电脑命令框输入“regedit”进入注册表 依次点击以下几个 右击新建一个“DWORD(32位)值”&#xff0c;命名为”FlightSettingsMaxPauseDays“ 右边选择十进制&#xff0c;左边填写暂停更新的天数 打开windows更新&#xff0c;进入高级选项 选择暂停更新的天数&#xff…

Fortigate防火墙二层接口的几种实现方式

初始配置 FortiGate出厂配置默认地址为192.168.1.99&#xff08;MGMT接口&#xff09;&#xff0c;可以通过https的方式进行web管理&#xff08;默认用户名admin&#xff0c;密码为空&#xff09;&#xff0c;不同型号设备用于管理的接口略有不同。 console接口的配置 防火墙…

4_1 Linux Centos7的RPM相关知识

4_1 Linux Centos7的RPM相关知识 文章目录 4_1 Linux Centos7的RPM相关知识1. RPM包文件名特征2. RPM软件包管理3. RPM包的一般安装位置4. RPM基本命令5. yum 系统环境centos7.9 临时挂载光盘文件&#xff1a;mount /dev/cdrom /dvd 1. RPM包文件名特征 * 软件名-版本信息.操作…

linux dtb解析方法之验证修改的dts是否有效

我们在调试linux系统外设的时候&#xff0c;经常修改dts文件&#xff0c;那怎么check修改是否生效呢 一、板载设备树路径 linux系统烧录完&#xff0c;登录系统后我们通常能看到如下的fdt和设备树文件&#xff1a; ls /sys/firmware/ devicetree fdt 二、解析板载设备树 那如…

oracle表锁

--oracle提醒记录被另一个用户锁住&#xff1a; --问题描述&#xff1a;你去修改数据时&#xff0c;报错“ --问题分析&#xff1a;你用select t.*,t.rowid from qxt_logsend_0728修改数据结果集时&#xff0c;计oracle会通过事务锁锁住这个记录&#xff0c;点击记录改变&#…

TypeScript常见面试题第十一节

题目五十一:什么是TypeScript映射文件? 一、讲解视频 TS面试题五十一:什么是TypeScript映射文件? 二、题目解析 本题目考察 ts 编译调试的基础内容,ts 的源映射文件通过编译选项–sourceMap 进行生成,指定后编译完了会自动生成对应映射文件,可以用于调试 ts 文件。 三…

java并发控制(猴子摘桃例子)

【问题】 有n个桃子&#xff0c; 猴子A每次固定摘2个&#xff0c;猴子B每次固定摘3个&#xff0c;这2只猴子不断摘桃子直到剩余桃子数量不足以摘&#xff08;必须满足摘桃个数&#xff09;&#xff1b; 【1】 使用AtomicInteger&#xff08;推荐&#xff09; 1&#xff09;利…