Exemplo n.º 1
0
def test_reshape_minus_1(t: Tensor) -> Tensor:
    return ep.reshape(t, -1)
Exemplo n.º 2
0
def test_reshape_int(t: Tensor) -> Tensor:
    n = 1
    for k in t.shape:
        n *= k
    return ep.reshape(t, n)
Exemplo n.º 3
0
def test_reshape(t: Tensor) -> Tensor:
    shape = (1, ) + t.shape + (1, )
    return ep.reshape(t, shape)