search

advancedresearch / poi

a pragmatic point-free theorem prover assistant
Apache License 2.0
136 stars 7 forks source link

Add `soft_arg_min` and `soft_arg_max` #984

Closed bvssvni closed 3 years ago

bvssvni commented 3 years ago

https://pytorch.org/docs/stable/nn.functional.html?highlight=softmax#torch.nn.functional.softmax