每天讲解一点PyTorch torch.matmul

Posted knowform

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了每天讲解一点PyTorch torch.matmul相关的知识,希望对你有一定的参考价值。

每天讲解一点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]])
>>> 

实现点乘功能:对应位相乘

以上是关于每天讲解一点PyTorch torch.matmul的主要内容,如果未能解决你的问题,请参考以下文章

每天讲解一点PyTorch F.softmax

每天讲解一点PyTorch torch.matmul

每天讲解一点PyTorch isinstance

每天讲解一点PyTorch isinstance

每天讲解一点PyTorch 12enumerate

每天讲解一点PyTorch 12enumerate