Function comp

imperative/python/test/unit/functional/test_tensor.py:409–410

(out, target_shp)

    data3 = Tensor(np.random.random(input3_shape).astype(np.float32))

    def comp(out, target_shp):
        assert out._tuple_shape == target_shp

    def func(x, shp):
        return F.reshape(x, shp)