tensor 的运算（加法、点乘、矩阵乘法）

To Shine2024-09-11 17:47

文章目录

- - 案例准备
  - - 加法
    - 点乘
    - 矩阵乘法
    - 注意

案例准备

python 复制代码

import torch

a = torch.tensor([[1,2,3],[4,5,6],[7,8,9]], dtype=torch.float)
b = torch.tensor([[10,10,10,],[10,10,10],[10,10,10,]], dtype=torch.float)

此时a,b为

加法

python 复制代码

a + 100

python 复制代码

a + b
a.add(b)

点乘

python 复制代码

a * b

矩阵乘法

python 复制代码

a @ b
a.matmul(b)
torch.mm(a,b)

注意

在pytorch中，如果一个运算方法后面加上下划线，代表就地改变原值，比如下面的a,这样做可以节省内存，但会直接改变a的值，请谨慎使用

python 复制代码

a.add_(b)