[英]Pad multiple torch tensor over the last dim
我有多個具有以下形狀的火炬張量
x1 = torch.Size([1, 512, 177])
x2 = torch.Size([1, 512, 250])
x3 = torch.Size([1, 512, 313])
我如何在最后一個維度上將所有這些張量填充 0,以具有像 ([1, 512, 350]) 這樣的獨特形狀。
我試圖做的是將它們轉換為 NumPy 數組並使用這兩行代碼:
if len(x1) < 350:
ff = np.pad(f, [(0, self.max_len - f.shape[0]), ], mode='constant')
f = ff
但不幸的是,它不會影響最后的暗淡,仍然,形狀不相等。 任何幫助將不勝感激謝謝
你可以簡單地做:
import torch.nn.functional as F
x = F.pad(x, (0, self.max_len - x.size(2)), "constant", 0)
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.