ConvLSTM2D in PyTorch

Are there any plans of adding ConvLSTM2D in PyTorch ?

5 Likes

Hi!

You might want to check the simple implementation provided in Passing hidden layers to ConvLSTM.

Cheers,
Petteri