Function custom_op_meta

intermediate_source/torch_export_tutorial.py:803–805

(x)

@custom_op.register_fake
def custom_op_meta(x):
    # Returns an empty tensor with the same shape as the expected output
    return torch.empty_like(x)

######################################################################
# Here is an example of exporting a program with the custom op.