pytorch中的unsqueeze()函數解析

0.doc文檔 doc文檔是這麼說的: 這裏簡單的譯一下。 1.func: 返回一個新的tensor,這個tensor 在指定的位置被插入了一個大小爲1的新維度 這個返回的tensor 和之前的 tensor 有着相同的數據 2.parameter 2.1 tensor: 就是需要被處理的那個tensor 2.2 dim 值的範圍在 [-input.dim()-1, input.dim()+1 )
相關文章
相關標籤/搜索