TensorPlay
API symbolsnn
Copy
View MarkdownDownload .md

tensorplay.nn.functional.hardshrink

tensorplay.nn.functional.hardshrink(input, lambd=0.5) → Tensor[source]

Applies the hard shrinkage function element-wise.

See Hardshrink for more details.

tensorplay.nn.functional.gumbel_softmax

Sample from the Gumbel-Softmax distribution and optionally discretize.

tensorplay.nn.functional.hardsigmoid

Next Page

Ask DeepWiki