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