    def test_tf_avgpool3d(self):
        """
        Test tf avgpool3d layer.
        """
        x=torch.randn(1,1,1,4,4)
        model = TfAvgPool3D()
        y=model(x)
        self.assertTrue(y.shape == torch.Size([1,1,1,2,2]))
Test tf avgpool3d layer.
tests/unittests/models/layers/test_tf_avgpool3d.py