I have some rnn model
   class Net(nn.Module):
def __init__(self):
super(Net, self).__init__()
# some code #
def forward(self, x):
x = self.rnn(x)
output = my_function(# here is a code to get input #)
return x, output

And I need use input in my function in this model. Сan i do it somehow