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