码迷,mamicode.com
首页 > 其他好文 > 详细

torch.unsqueeze

时间:2019-01-23 18:13:15      阅读:223      评论:0      收藏:0      [点我收藏+]

标签:pre   width   指定位置   info   http   ima   基础   idt   返回   

unsqueeze在指定位置处插入尺寸为1的维度。

注意:返回的张量与此张量共享相同的基础数据。

技术分享图片

Example:

>>> x = torch.tensor([1, 2, 3, 4])
>>> torch.unsqueeze(x, 0)
tensor([[ 1,  2,  3,  4]])
>>> torch.unsqueeze(x, 1)
tensor([[ 1],
        [ 2],
        [ 3],
        [ 4]])

 

torch.unsqueeze

标签:pre   width   指定位置   info   http   ima   基础   idt   返回   

原文地址:https://www.cnblogs.com/jiangkejie/p/10310400.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!