Summary of the model which includes GRU layers

Hello,
I would like to represent my model in a way similar to summary in keras.
I know this is not new question but the thing that I am using GRU layers and using summary from
from torchsummary import summary
get me this error
TypeError: forward() missing 1 required positional argument: 'hidden'
however I sending it ( 1, hidden, number of features) as the input ?

Please advice
Abeer