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