search

pytorch / pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration
https://pytorch.org
Other
82.67k stars 22.26k forks source link

missing docs for torch.Tag #126518

Open ngc92 opened 4 months ago

ngc92 commented 4 months ago

📚 The doc issue

The documentation for torch.Tag is virtually non-existent. This seems particularly glaring, given that torch.library.define explicitly asks users to carefully read those docs.

Suggest a potential alternative/fix

No response

cc @svekars @brycebortree @anjali411

mikaylagawarecki commented 4 months ago

cc @zou3519 since you added the tags argument to torch.library.define