Function torch::nn::functional::cross_entropy¶
Defined in File loss.h
Function Documentation¶
-
inline Tensor torch::nn::functional::cross_entropy(const Tensor &input, const Tensor &target, const CrossEntropyFuncOptions &options = {})¶
See https://pytorch.org/docs/main/nn.functional.html#torch.nn.functional.cross_entropy about the exact behavior of this functional.
See the documentation for
torch::nn::functional::CrossEntropyFuncOptions
class to learn what optional arguments are supported for this functional.Example:
namespace F = torch::nn::functional; F::cross_entropy(input, target, F::CrossEntropyFuncOptions().ignore_index(-100).reduction(torch::kMean));