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