Ejemplo n.º 1
0
def test_min_none_keepdims(t: Tensor) -> Tensor:
    return ep.min(t, axis=None, keepdims=True)
Ejemplo n.º 2
0
def test_min_keepdims(t: Tensor) -> Tensor:
    return ep.min(t, axis=0, keepdims=True)
Ejemplo n.º 3
0
def test_min_axis(t: Tensor) -> Tensor:
    return ep.min(t, axis=0)
Ejemplo n.º 4
0
def test_min_axes(dummy: Tensor) -> Tensor:
    t = ep.ones(dummy, 30).float32().reshape((3, 5, 2))
    return ep.min(t, axis=(0, 1))
Ejemplo n.º 5
0
def test_min(t: Tensor) -> Tensor:
    return ep.min(t)