Typedef torch::nn::functional::BinaryCrossEntropyWithLogitsFuncOptions

• Defined in File loss.h

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