Function MakeTallFunc

dali/test/python/operator_2/test_reshape.py:99–107

(relative, wildcard)

def MakeTallFunc(relative, wildcard):
    def func(image):
        if relative:
            return np.array([-1 if wildcard else 2, 0.5, 1]).astype(np.float32)
        else:
            h, w, c = image.shape
            return np.array([-1 if wildcard else 2 * h, w / 2, c]).astype(np.int32)

    return func


class ReshapeWithArgInput(Pipeline):