Thanks for your reply! I just want to forward batches. I know people usually rewrite the forward method and derive the result by
out = net(input)
I add the forward_batch method because the inputs to my network sometimes are different. I derive the result by
out = net.forward_batch(input)
It seems that the returned class doesn’t support this method. Maybe I should change the name of the method to forward?