I am new to PyTorch, thanks for reply.
This seems the definition of RNNCell 64,  Suppose I write this:
class ConvLSTMCell(nn.Module):
do you mean that in this ConvLSTMCell module, I can use nn.Linear and nn.conv2d , and it’s with autorgrad function?