在 Pytorch 中, 点积由 torch.mul 实现, 矩阵相乘由 torch.mm 完成

    import torch
    
    input1 = torch.Tensor([[1,2],[3,4]])
    input2 = torch.Tensor([[2,2],[2,2]])
    print(input1.shape)
    print(input2.shape)
    output = torch.mul(input1,input2)
    print(output)
    print(output.shape)
    
    
      
      
      
      
      
      
      
      
      
    

    torch.Size([2, 2])
    torch.Size([2, 2])
    tensor([[2., 4.],
        [6., 8.]])    # 看结果
    torch.Size([2, 2])