    def test_temporal_cg_avgpool3d(self):
        """
        Test temporal cg avgpool3d layer.
        """
        x=torch.randn(1,1,1,4,4)
        model = TemporalCGAvgPool3D()
        y=model(x)
        self.assertTrue(y.shape == torch.Size([1,1,1,4,4]))
Test temporal cg avgpool3d layer.
tests/unittests/models/layers/test_temporal_cg_avgpool3d.py