PyTorch_阿达玛积

阿达玛积指的是矩阵对应位置的元素相乘，可以使用乘号运算符，也可以使用mul函数来完成计算。

---

代码

import torch 
import numpy as np 

# 1. 使用 mul 函数
def test01():
    data1 = torch.tensor([[1, 2], [3, 4]])
    data2 = torch.tensor([[5, 6], [7, 8]])

    data = data1.mul(data2) 
    print(data) 

# 2. 使用 * 号运算符
def test02():
    data1 = torch.tensor([[1, 2], [3, 4]])
    data2 = torch.tensor([[5, 6], [7, 8]])

    data = data1 * data2 
    print(data)

if __name__ == "__main__":
    test01()
    test02()