每天讲解一点PyTorch 【1】torch.matmul

cv.exp

于 2021-11-12 23:33:24 发布

阅读量181

点赞数

CC 4.0 BY-SA版权

分类专栏： Deep Learning 文章标签： pytorch 深度学习机器学习

本文链接：https://blog.csdn.net/forest_world/article/details/121297377

Deep Learning 专栏收录该内容

172 篇文章

订阅专栏

本文详细介绍了PyTorch中的torch.matmul函数，它用于执行张量的矩阵乘法。通过示例展示了如何使用该函数进行两个二维张量的乘法操作，演示了不同形状张量之间的乘法运算，以及点乘（对应位置相乘）的区别。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

每天讲解一点PyTorch——函数torch.matmul

torch.matmul

今天我们学习函数torch.matmul：Tensor的乘法

// An highlighted block
    >>> import torch
>>> x = torch.rand(2,2)
>>> x
tensor([[0.7834, 0.5647],
        [0.2723, 0.6277]])
>>> y = torch.rand(2,2)
>>> y
tensor([[0.3738, 0.1724],
        [0.3732, 0.3012]])
>>> 
>>> z = torch.rand(1,2)
>>> z
tensor([[0.8670, 0.3807]])
>>> 
>>> torch.mul(x,y)
tensor([[0.2928, 0.0973],
        [0.1016, 0.1891]])

>>> m = torch.tensor([[1,2],[3,4]])
>>> m
tensor([[1, 2],
        [3, 4]])
>>> 
>>> n = torch.tensor([[2,3],[4,5]])
>>> n
tensor([[2, 3],
        [4, 5]])
>>> 
>>> torch.mul(m,n)
tensor([[ 2,  6],
        [12, 20]])
>>> 
>>> torch.mul(m,m)
tensor([[ 1,  4],
        [ 9, 16]])
>>> 
>>> torch.mul(n,n)
tensor([[ 4,  9],
        [16, 25]])
>>>