Method get_dim

tools/utils/layers.py:239–240

(self)

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

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

    def forward(self, motion_v: torch.Tensor) -> torch.Tensor:
        """