基础
加减法
遇到不匹配情况,会自动进行broadcasting进行维度匹配后,再进行操作
运算符得到了重载
乘法
Torch.mm():仅在2维中使用
Torch.matmul()
@重载后的乘法操作符
即使是多维矩阵相乘,实际相乘的仍然是最后两维数据,所以前后行列向量维数必须匹配
当其他维度不匹配时,仍然使用Broadingcast进行维度匹配后运算
Power&sqrt
.rsqrt():平方根的倒数
对数、指数
近似值
floor():向下取整
ceil():向上取整
trunc():截取整数位
frac():截取小数位
round():四舍五入
clamp
一个参数,参数为最小值
两个参数,表示(最小值,最大值)