【PyTorch】 tensor.squeeze壓縮維度

• 1、背景
• 2、squeeze方法
• 3、示例

1、背景

pytorch的tensor維度擴展、壓縮是一個常見的操作，
tensor維度擴展參見如下博客：
【PyTorch】 tensor擴展一個維度
https://jn10010537.blog.csdn.net/article/details/120303933

本博客tensor壓縮一個維度的方法：squeeze，即壓縮。
（除了壓縮一個維度，還支持將所有為1的維度壓縮掉，具體參見示例）

關于squeeze方法的說明如下：

import torch print(torch.__version__) help(torch

總結

以上是生活随笔為你收集整理的【PyTorch】 tensor.squeeze压缩维度的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。