def output_type(self):
    value_dtype = self.arg.type().value_type
    return rlz.shape_like(self.arg, value_dtype)
def output_type(self):
    value_dtype = self.arg.type().value_type
    return rlz.shape_like(self.arg, value_dtype)