1分鐘理解pytorch的reshape函數中-1表示的意義

先說答案,reshape函數中-1代表的是n,什麼意思呢,函數中另一個參數決定了-1的值,看下面三張圖就很容易理解了

  1. 定義34的張量,reshape(-1, 1),你把它想象成要轉換成n1的矩陣,那是多少?沒錯,最後轉換成了12*1的矩陣
    在這裏插入圖片描述

  2. 同理,如果是reshape(1, -1)呢?那就是1*12了
    在這裏插入圖片描述

  3. 最後的一個更加直觀,如果是reshape(2, -1)呢?沒錯,那就是2*6的矩陣了,是不是1分鐘就瞭解了?在這裏插入圖片描述

總結

-1代表n
n=tensor的長度/第一個參數

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章