Add IGamma operator

4 years ago · 5e9178c5b6
parent 52eb2d3401
commit 5e9178c5b6
2 changed files with 268 additions and 9 deletions
--- a/mindspore/nn/layer/math.py
+++ b/mindspore/nn/layer/math.py
--- a/tests/ut/python/ops/test_nn_ops.py
+++ b/tests/ut/python/ops/test_nn_ops.py
@ -593,6 +593,11 @@ test_cases = [
        'block': nn.LGamma(),
        'desc_inputs': [Tensor(np.array([3, 4, 5, 6]).astype(np.float32))],
        'skip': ['backward']}),
    ('IGamma', {
        'block': nn.IGamma(),
        'desc_inputs': [Tensor(np.array([3, 4, 5, 6]).astype(np.float32)),
                        Tensor(np.array([3, 4, 5, 6]).astype(np.float32))],
        'skip': ['backward']}),
    ('FlattenNet', {
        'block': FlattenNet(),
        'desc_inputs': [Tensor(np.ones([1, 2, 3, 4], np.float32))],