五、Pytorch tensor 的数学运算

目录

1、基础运算

（1）加法运算

（2）减法运算

（3）哈达玛积(element wise，对应元素相乘)

（4）除法运算

2、矩阵运算

（1）二维矩阵相乘

（2）多维矩阵相乘

3、幂运算

4、开方运算

5、指数与对数运算

6、近似值运算

7、裁剪运算

---

1、基础运算

• 可以使用 + - * / 推荐
• 也可以使用 torch.add, mul, sub, div

（1）加法运算

def add():
    # add +
    # 这两个Tensor加减乘除会对b自动进行Broadcasting
    a = torch.rand(3,4)
    b = torch.rand(4)
    print("a = {}".format(a))
    print("b = {}".format(b))
    # a、b列数相同