tensor 的运算（加法、点乘、矩阵乘法）
文章目录
- 案例准备
- 加法
- 点乘
- 矩阵乘法
- 注意
案例准备
import torcha = 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)
