noise = noise.data.reshape(data.size()[0],input_dim,1,1).normal_(0,1) RuntimeError: shape '[32, 100,

pytorch 報錯noise =noise.data.reshape(data.size()[0],input_dim,1,1).normal_(0,1)RuntimeError: shape ‘[32, 100, 1, 1]’ is invalid for input of size 6400 緣由: 維度寫錯了 解決辦法: 因爲我是dataloader加載的數據,因此加上屬性drop_las
相關文章
相關標籤/搜索