我有多個火炬張量,形狀如下
x1 = torch.Size([1, 512, 177])
x2 = torch.Size([1, 512, 250])
x3 = torch.Size([1, 512, 313])
如何在最后一個維度上用0填充所有這些張量,使其具有一個獨特的形狀,如([1512350])。
我試圖將它們轉換為NumPy數組,并使用以下兩行代碼:
if len(x1) < 350:
ff = np.pad(f, [(0, self.max_len - f.shape[0]), ], mode='constant')
f = ff
但不幸的是,它不會影響最后的暗淡,而且形狀不相等。任何幫助都將不勝感激,謝謝
您可以簡單地執行以下操作: