Function torch::nn::functional::gumbel_softmax

• Defined in File activation.h

Function Documentation

inline Tensor torch::nn::functional::gumbel_softmax(const Tensor &logits, const GumbelSoftmaxFuncOptions &options = {})

See https://pytorch.org/docs/main/nn.functional.html#torch.nn.functional.gumbel_softmax about the exact behavior of this functional.

See the documentation for torch::nn::functional::GumbelSoftmaxFuncOptions class to learn what optional arguments are supported for this functional.

Example:

namespace F = torch::nn::functional;
F::gumbel_softmax(logits, F::GumbelSoftmaxFuncOptions().hard(true).dim(-1));