pytorch - 填充张量直到达到所需的大小

Question

我正在使用形状为 (X,42) 的 certian 张量，而 X 可以在 50 到 70 之间的范围内。
我想填充我得到的每个张量，直到它达到 70 的大小。所以所有张量都是 (70,42)。
当我的起始大小是一个变量 X 时，有没有办法做到这一点？谢谢您的帮助!

Answer 1

使用 torch.nn.functional.pad - 垫张量。

import torch
import torch.nn.functional as F

source = torch.rand((3,42))
source.shape
>>> torch.Size([3, 42])

# here, pad = (padding_left, padding_right, padding_top, padding_bottom)
source_pad = F.pad(source, pad=(0, 0, 0, 70 - source.shape[0]))

source_pad.shape
>>> torch.Size([70, 42])