torch.Tensor常用操作 |
您所在的位置:网站首页 › torchtensor的功能 › torch.Tensor常用操作 |
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 |