main
forward
def forward(self, x): mapped = self.net(x) batch_size = mapped.size(0) time_length = mapped.size(1)