Tensors and Their Operations

约 113 个字 预计阅读时间不到 1 分钟

Shape Operations

squeeze & unsqueeze

• torch.squeeze(input: Tensor, dim: Optional[int, List[int]] = None) -> Tensor
• torch.unsqueeze(input: Tensor, dim: int) -> Tensor

torch.squeeze 移除张量形状中为 1 的维度/Singleton Dimensions，比如将形状为 (1, 3, 1, 5) 的张量变为 (3, 5)。添加参数 torch.squeeze(dim) 可以指定移除特定参数。但是，其返回的张量和输入的张量共享存储，因此改变其中一个的内容会改变另外一个的内容。

`