机器学习PyTorch——1.基本运算

目录

一.加减乘除

二.矩阵相乘

三.其它运算

四.取整运算

五.clamp函数，限制最大最小值

---

一.加减乘除

（1）在pytorch中，既可以使用基本的：+，-，*，/，运算符号进行计算，也可以使用pytorch的内置计算函数：add，sub，mul，div。需要注意的是，pytorch中的tensor再进行计算时，会自动进行维度的转换。例如：

这里在进行加运算前，将b已经进行维度转换为[3,4]。

（2）我们这里来看一下内置函数的使用：

这里我们需要注意，这里的运算结果，和直接使用符号进行运算是一样的，我们可以用 eq（）来进行一下验证：

二.矩阵相乘

（1）使用 torch.mm（）