You can pad extra elements like so: import torch.nn.functional as F n = self.batchsize - b new_x = F.pad(x, ... How to pad zeros on Batch, PyTorch.
確定! 回上一頁