每天讲解一点PyTorch torch.matmul

Posted 2021-12-11 knowform

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

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