Method __init__

tools/utils/layers.py:232–237

(self, cfg: DictConfig, channel_dim: int, motion_v_dim: int)

@persistence.persistent_class
class TemporalInput(nn.Module):
    def __init__(self, cfg: DictConfig, channel_dim: int, motion_v_dim: int):
        super().__init__()

        self.cfg = cfg
        self.motion_v_dim = motion_v_dim
        self.const = nn.Parameter(torch.randn(1, channel_dim, 4, 4))

    def get_dim(self):
        return self.motion_v_dim + self.const.shape[1]