File size: 285 Bytes
62dca4c | 1 2 3 4 5 6 7 8 9 10 11 12 | import torch
@torch.no_grad()
def padding(tensor, left=True):
zeropadding = torch.zeros_like(tensor[:, -1:])
if left:
tensor = torch.cat((zeropadding, tensor[:, :-1]), dim=1)
else:
tensor = torch.cat((tensor[:, 1:], zeropadding), dim=1)
return tensor
|