Esempio n. 1
0
def test_prod_bool(t: Tensor) -> Tensor:
    return ep.prod(t != 0)
Esempio n. 2
0
def test_prod_int(t: Tensor) -> Tensor:
    return ep.prod(ep.arange(t, 5))
Esempio n. 3
0
def test_prod_none_keepdims(t: Tensor) -> Tensor:
    return ep.prod(t, axis=None, keepdims=True)
Esempio n. 4
0
def test_prod_keepdims(t: Tensor) -> Tensor:
    return ep.prod(t, axis=0, keepdims=True)
Esempio n. 5
0
def test_prod_axes(dummy: Tensor) -> Tensor:
    t = ep.ones(dummy, 30).float32().reshape((3, 5, 2))
    return ep.prod(t, axis=(0, 1))
Esempio n. 6
0
def test_prod_axis(t: Tensor) -> Tensor:
    return ep.prod(t, axis=0)
Esempio n. 7
0
def test_prod(t: Tensor) -> Tensor:
    return ep.prod(t)