I use lstm from pytorch in my code to predict time series. while i write this code
class LSTM_model(nn.Module):
def __init__(self, input_size, outp