张量的加法、减法、乘法和矩阵乘法
最编程
2024-05-22 14:03:11
...
import torch
"""如下是tensor乘法与加减法，对应位相乘或相加减，可以一对多"""
def add_and_mul():
x = torch.Tensor([ [[1,2,3],
[4,5,6]],
[[7,8,9],
[10,11, 12]] ])
y= torch.Tensor([1,2,3])
y=y-x
t=1. - x.sum(dim=1)
y= torch.Tensor([[1,2,3],
[4,5,6]])
y=y*x
z=x**2
print(y)
print(z)
"""如下为矩阵乘法"""
def matmul_and_bmm():
#a=(2*3*4)
a = torch.Tensor([ [[1,2,3,4],
[4,0,6,0],
[3,2,1,4]],
[[3,2,1,0],
[0,3, 2,2],
[1, 2, 1,0]] ])
#b=(2,2,4)
b = torch.Tensor([ [[1,2,3,4],
[4,0,6,0]],
[[3,2,1,0],
[1, 2, 1,0]] ])
#res=(2,3,2),对于a*b，是第一维度不变，而后3x4 * 4x2=3x2
res=torch.matmul(a,b.transpose(1,2))#维度res=[2,3,2]
res2 = torch.bmm(a, b.transpose(1, 2))#维度res2=[2,3,2]
print(res)#res2的值等于res
"""
tensor([[[30., 22.],
[22., 52.],
[26., 18.]],
[[14., 8.],
[ 8., 8.],
[ 8., 6.]]])
"""
上一篇： excel 如何使用宏
下一篇： 矩阵加减法算法 - Nuggets
推荐阅读
-
批处理功能：加法、减法、大数乘法和除法、周期计算、数字排序、转换等。
-
向量和矩阵的点叉乘法 - Numpy 实现向量的点叉乘法
-
矩阵和向量叉乘法的推导和简单应用
-
数学基础 - 矩阵的点乘法和叉乘法
-
excel 函数公式加法、减法、乘法和除法
-
sql 加法、减法、乘法和除法
-
SQL Server 加法、减法、乘法、除法、四则运算、比较、逻辑和并集操作（SQL Server 2008） 5 - SQL Server 2008 中的四则运算、比较运算、逻辑运算和连接运算
-
pandas 数据处理基础 - 基本加法、减法、乘法和除法运算规则
-
用 Excel 创建随机加法、减法、乘法和除法练习题
-
Excel 快速入门教程 2 - 加法、减法、乘法和除法