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