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()