There is also this model:

  
      github.com
  
  
    Atcold/pytorch-CortexNet/blob/master/model/ConvLSTMCell.py 1.1k
import torch
from torch import nn
import torch.nn.functional as f
from torch.autograd import Variable


# Define some constants
KERNEL_SIZE = 3
PADDING = KERNEL_SIZE // 2


class ConvLSTMCell(nn.Module):
    """
    Generate a convolutional LSTM cell
    """

    def __init__(self, input_size, hidden_size):
        super().__init__()
        self.input_size = input_size
        self.hidden_size = hidden_size


  This file has been truncated. show original