栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

关于pytorch的张量处理

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

关于pytorch的张量处理

1. squeeze()

压缩张量,去除size为1的维度

2.unsqueeze()

增加维度:在指定维度上增加维度,

如b = torch.tensor([1, 2, 3]), torch.Size([3])

在dim = 0的维度插入,B1 = torch.unsqueeze(b, 0),输出为[[1,2,3]]。torch.Size([1, 3])

加到另一个维度,B2 = B.unsqueeze(1),输出为[[1],[2],[3]]。torch.Size([3, 1])

在B1基础上加新的维度,B3 = B1.unsqueeze(2), 输出为[[[1],[2],[3]]]。torch.Size([1, 3, 1])

3.expand()

在指定维度上进行扩张

4.gather()

从tensor中按索引取值,注意!索引范围是[0, length - 1],如果范围超出则会报错!

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/925889.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号