(tensor, left=True)
| 462 | |
| 463 | @torch.no_grad() |
| 464 | def padding(tensor, left=True): |
| 465 | zeropadding = torch.zeros_like(tensor[:, -1:]) |
| 466 | if left: |
| 467 | tensor = torch.cat((zeropadding, tensor[:, :-1]), dim=1) |
| 468 | else: |
| 469 | tensor = torch.cat((tensor[:, 1:], zeropadding), dim=1) |
| 470 | return tensor |
| 471 | |
| 472 | |
| 473 |