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));