pytorch矩阵乘法

*

# 对应元素相乘
a * b

.mul

# 对应元素相乘
torch.mul(a, b)

.mm

# 矩阵相乘
torch.mm(a, b)

.matmul

# 矩阵相乘
torch.matmul(a,b)

你可能感兴趣的:(python)