Typedef torch::nn::functional::CrossEntropyFuncOptions¶
Defined in File loss.h
Typedef Documentation¶
-
using
torch::nn::functional
::
CrossEntropyFuncOptions
= CrossEntropyLossOptions¶ Options for
torch::nn::functional::cross_entropy
.See the documentation for
torch::nn::CrossEntropyLossOptions
class to learn what arguments are supported.Example:
namespace F = torch::nn::functional; F::cross_entropy(input, target, F::CrossEntropyFuncOptions().ignore_index(-100).reduction(torch::kMean));