pytorch
253ffbf2 - Exposing native _scaled_dot_product_attention to torch.nn (#85044)

Commit
2 years ago
Exposing native _scaled_dot_product_attention to torch.nn (#85044) # Summary This exposes the _scaled_dot_product_attention function to python in the nn namespace. It is still underscored because the api for args, and kwargs is still in flux for the next few weeks and will eventually land as a prototype feature. Pull Request resolved: https://github.com/pytorch/pytorch/pull/85044 Approved by: https://github.com/cpuhrsch
Author
Committer
Parents
Loading