torch 运算符号总结

目录

点乘：

@ 矩阵乘法

点乘和矩阵相乘

---

点乘：

PyTorch中点乘与矩阵操作详解-优快云博客

@ 矩阵乘法

import torch

# 创建两个示例张量
A = torch.tensor([[1, 2], [3, 4]])
B = torch.tensor([[5, 6], [7, 8]])

# 使用 @ 运算符进行矩阵乘法
result = A @ B
print(result)

点乘和矩阵相乘

PyTorch中点乘与矩阵操作详解-优快云博客