Как использовать вход в модели RNN

Question

У меня есть модель RSN

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

И мне нужно использовать вход в моей функции в этой модели. Могу ли я сделать это как-то