IgANet
IgANets - Isogeometric Analysis Networks
Loading...
Searching...
No Matches
iganet::GumbelSoftmax Member List

This is the complete list of members for iganet::GumbelSoftmax, including all inherited members.

apply(const torch::Tensor &input) const overrideiganet::GumbelSoftmaxinlinevirtual
GumbelSoftmax(torch::nn::functional::GumbelSoftmaxFuncOptions options={})iganet::GumbelSoftmaxinlineexplicit
GumbelSoftmax(double tau, int dim, bool hard)iganet::GumbelSoftmaxinlineexplicit
name() const noexceptiganet::utils::FullQualifiedNameinlineprotectedvirtual
name_iganet::utils::FullQualifiedNamemutableprotected
options() constiganet::GumbelSoftmaxinline
options()iganet::GumbelSoftmaxinline
options_iganet::GumbelSoftmaxprivate
pretty_print(std::ostream &os=Log(log::info)) const noexcept overrideiganet::GumbelSoftmaxinlinevirtual
read(torch::serialize::InputArchive &archive, const std::string &key="gumbel_softmax") overrideiganet::GumbelSoftmaxinlinevirtual
write(torch::serialize::OutputArchive &archive, const std::string &key="gumbel_softmax") const overrideiganet::GumbelSoftmaxinlinevirtual
~ActivationFunction()=defaultiganet::ActivationFunctionvirtual
~GumbelSoftmax() override=defaultiganet::GumbelSoftmax