Generalized function signature for models.

[ErnstRoell]

Request change in the function signature of forward for the models since the toponetx models have a different signature and that would yield issues down the road.

Changes:

• Manually calculate the batch length (either in batch.batch or last arg of batch)
• change signature of forward to def forward(self, batch)