深度学习笔记（4）——维度操作（1）压缩维度与扩展维度

深度学习笔记（4）——维度操作（1）压缩维度与扩展维度

---

前言

维度操作在深度学习的模型搭建的过程中是一类常用的操作，其中对维度的操作有增加维度、压缩维度、展平、·拼接、拆分、维度重塑 ，转置’,维度交换等，下面总结下目前可以想到的维度操作。

一、扩展维度

1.1.1 torch.unsqueeze(input,dim)

torch.unsqueeze(input,dim)
input：要操作的tensor
dim：需要加的维度

1.1.2实验

import torch
x = torch.tensor([[7,8,9,10],[5,6,7,8]])
y = torch.unsqueeze(x,2)
print("x.shape:",x.shape)
print("x:",x)
print("y.shape:",y.shape)
print("y:",y)

实验结果：

x.shape: torch.Size(