torch.Tensor常用操作

您所在的位置:网站首页 torchtensor的功能 torch.Tensor常用操作

torch.Tensor常用操作

#torch.Tensor常用操作| 来源: 网络整理| 查看: 265

1) tensor的乘法

a)torch.mm

b)torch.matmul

c)torch.mul

d)使用*操作法

torch.mul和*操作符号是一致的,都是点乘。

torch.mm是矩阵乘法,比如m*k与k*n矩阵相乘,结果是m*n

torch.matmul是torch.mm的broadcase版本,可以处理第一维度为batch_size的乘法

2) torch的维度的相互转换

torch.transpose()

torch.tranpose(dim1, dim2)把dim1和dim2进行互换

3)torch的常用函数

torch.exp

torch.log

torch.log2

torch.floor() 向下取整

torch.ceil() 向上取整

torch.tunc() 取整数部分

torch.frac() 取小数部分

torch.round() 四舍五入部分

5) torch按照维度取大,取小等

torch.max(-1) 按照矩阵的最后一个维度, 使用values取值,使用indices取指标

6)torch.tensor.view

这个就是按照多少大小取看这个矩阵,和原来的reshape是一样的用法

7)torch.tensor squeeze和unsqueeze的使用

去除一些无效的dim=1,和补充一些dim=1的维度



【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3