Thanks, but where to call this function?
If self is a nn.Module, should we call it in __init__() function or forward() function? Should we call it before or after self.lstm1(input, hidden)? I always get illegal memory access errors