Skip to content

Deep learning

Deep learning is an AI function and subset of machine learning, used for processing large amounts of complex data.

Here are 19,763 public repositories matching this topic...

yf225
yf225 commented Sep 30, 2019

Context

We would like to add torch::nn::functional::gumbel_softmax to the C++ API, so that C++ users can easily find the equivalent of Python API torch.nn.functional.gumbel_softmax.

Steps

  • Add torch::nn::GumbelSoftmaxOptions to torch/csrc/api/include/torch/nn/options/activation.h (add this file if it doesn’t exist), which should include the following parameters (based on
You can’t perform that action at this time.