Pytorch入门（番外）点乘与相乘

原创于 2019-12-10 18:54:51 发布 · 7.3k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #python #variable

Pytorch 专栏收录该内容

6 篇文章

订阅专栏

本文通过实例解析了PyTorch中矩阵乘法的正确实现方式，区分了点乘与矩阵乘法的不同，并展示了如何使用torch.mm()函数进行矩阵乘法运算。

部署运行你感兴趣的模型镜像

今天看到一行pytorch的代码

import torch
from torch.autograd import Variable

tensor = torch.FloatTensor([[1,2],[3,4]])          
variable = Variable(tensor, requires_grad=True) 
v_out = torch.mean(variable*variable)

很理所当然的理解为两个矩阵相乘，但是打印输出看的时候觉得不对

tensor([[ 1.,  4.],
        [ 9., 16.]], grad_fn=<MulBackward0>)

这里明显做了一个点乘

那如何才能让这两个variable变量做矩阵的乘法呢

print(torch.mm(variable,variable))

tensor([[ 7., 10.],
        [15., 22.]], grad_fn=<MmBackward>)

嗯，这样就可以了

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch

Cuda

PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理