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

先說答案,reshape函數中-1代表的是n,什麼意思呢,函數中另一個參數決定了-1的值,看下面三張圖就很容易理解了 定義34的張量,reshape(-1, 1),你把它想象成要轉換成n1的矩陣,那是多少?沒錯,最後轉換成了12*1的矩陣 同理,如果是reshape(1, -1)呢?那就是1*12了 最後的一個更加直觀,如果是reshape(2, -1)呢?沒錯,那就是2*6的矩陣了,是不是1分鐘
相關文章
相關標籤/搜索