Table of Contents
Shortcuts

# torch.utils.generate_methods_for_privateuse1_backend

torch.utils.generate_methods_for_privateuse1_backend(for_tensor, for_module, for_storage, unsupported_dtype) → None

  Automatically generate attributes and methods for the custom backend after rename privateuse1 backend. In the default scenario, storage-related methods will not be generated automatically.

  When you implement kernels for various torch operations, and register them to the PrivateUse1 dispatch key. And call the function torch.rename_privateuse1_backend(“foo”) to rename your backend name. At this point, you can easily register specific methods and attributes by calling this function. Just like torch.Tensor.foo(), torch.Tensor.is_foo, torch.Storage.foo(), torch.Storage.is_foo.

  Note: We recommend you use generic functions (check devices are equal or to(device=)). We provide these methods for convenience only and they will be “monkey patched” onto the objects and so will not be properly typed. For Storage methods generate, if you need to support sparse data storage, you need to extend the implementation yourself.

  Parameters:

    - for_tensor (bool) – whether register related methods for torch.Tensor class.

    - for_module (bool) – whether register related methods for torch.nn.Module class.

    - for_storage (bool) – whether register related methods for torch.Storage class.

    - unsupported_dtype (List[torch.dtype]) – takes effect only when the storage method needs to be generated, indicating that the storage does not support the torch.dtype type.

  Example:

      >>> torch.utils.rename_privateuse1_backend("foo")
      >>> torch.utils.generate_methods_for_privateuse1_backend()
      # Then automatically generate backend-related attributes and methods.
      >>> a = torch.tensor(2).foo()
      >>> a.is_foo
      >>> hasattr(torch.nn.Module, 'foo')

## Docs

Access comprehensive developer documentation for PyTorch

View Docs

## Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

## Resources

Find development resources and get your questions answered

View Resources

To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. As the current maintainers of this site, Facebook’s Cookies Policy applies. Learn more, including about available controls: Cookies Policy.