示例#1
0
 def rshift_zeroPad(self, x):
     return torch.cat((ftens(x.size(0), x.size(1), 1, x.size(3),
                             x.size(4)).fill_(0), x[:, :, :-1]),
                      dim=2)
示例#2
0
 def lshift_zeroPad(self, x):
     return torch.cat(
         (x[:, :, 1:], ftens(x.size(0), x.size(1), 1, x.size(3),
                             x.size(4)).fill_(0)),
         dim=2)