深入浅出Pytorch函数——torch.unsqueeze

分类目录:《深入浅出Pytorch函数》总目录

相关文章:

· 深入浅出Pytorch函数------torch.squeeze

· 深入浅出Pytorch函数------torch.unsqueeze


返回一个新的张量,且在指定位置新插入了一个大小为1的维度,返回的张量与该张量共享相同的底层数据。

语法

c 复制代码
torch.unsqueeze(input, dim) → Tensor

参数

  • input:[Tensor] 输入张量
  • dim:[int] 插入维度的位置索引,范围是 [ -input.dim() − 1 , input.dim() + 1 ] [\text{-input.dim()} - 1, \text{input.dim()} + 1] [-input.dim()−1,input.dim()+1],默认值为 input.dim() + 1 \text{input.dim()} + 1 input.dim()+1

实例

输入:

复制代码
x = torch.tensor([1, 2, 3, 4])
torch.unsqueeze(x, 0)

输出:

复制代码
tensor([[ 1,  2,  3,  4]])

输入:

复制代码
x = torch.tensor([1, 2, 3, 4])
torch.unsqueeze(x, 1)

输出:

复制代码
tensor([[1],
        [2],
        [3],
        [4]])
相关推荐
加油吧zkf18 分钟前
Conda虚拟环境管理:从入门到精通的常用命令
图像处理·深度学习·计算机视觉·conda
算家计算1 小时前
“28项评测23项SOTA——GLM-4.1V-9B-Thinking本地部署教程:10B级视觉语言模型的性能天花板!
人工智能·开源
Codebee1 小时前
OneCode注解驱动:智能送货单系统的AI原生实现
人工智能·低代码
2401_878624791 小时前
pytorch 自动微分
人工智能·pytorch·python·机器学习
胖达不服输1 小时前
「日拱一码」021 机器学习——特征工程
人工智能·python·机器学习·特征工程
Rvelamen1 小时前
大模型安全风险与防护产品综述 —— 以 Otter LLM Guard 为例
人工智能
MARS_AI_1 小时前
大语言模型驱动智能语音应答:技术演进与架构革新
人工智能·语言模型·自然语言处理·架构·信息与通信
程序员小灰1 小时前
AI独角兽团队Manus裁员80人,剩下40人迁至新加坡总部!
人工智能·aigc·agent
新智元1 小时前
OpenAI去年挖的坑填上了!奖励模型首现Scaling Law,1.8B给70B巨兽上了一课
人工智能·openai