示例#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)