Rate this Page

Typedef torch::nn::functional::BinaryCrossEntropyWithLogitsFuncOptions#

Typedef Documentation#

using torch::nn::functional::BinaryCrossEntropyWithLogitsFuncOptions = BCEWithLogitsLossOptions#

Options for torch::nn::functional::binary_cross_entropy_with_logits.

See the documentation for torch::nn::BCEWithLogitsLossOptions class to learn what arguments are supported.

Example:

namespace F = torch::nn::functional;
F::binary_cross_entropy_with_logits(input, target,
F::BinaryCrossEntropyWithLogitsFuncOptions().pos_weight(pos_weight).reduction(torch::kSum));