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

文章目录

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

案例准备

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为

加法

a + 100

a + b
a.add(b)

点乘

a * b

矩阵乘法

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

注意

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

a.add_(b)