Ah, got it! Thank you. It would be nice though is this things were automatically taken care for instance in a sequential model. May I ask another question not completely related to the above one but since I’m not a pro pytorch user maybe you might know. Does pytorch offer dynamic unrolling of inputs? something equivalent to tensorflow’s tf.nn.dynamic_rnn?