Class BinaryCrossEntropy

• Defined in File binary_cross_entropy.h

Inheritance Relationships

Base Type

• public mindspore::ops::PrimitiveC (Class PrimitiveC)

Class Documentation

class BinaryCrossEntropy : public mindspore::ops::PrimitiveC

Computes the binary cross entropy between the logits and the labels. Refer to Python API mindspore.ops.BinaryCrossEntropy for more details.

Public Functions

inline BinaryCrossEntropy()

Constructor.

~BinaryCrossEntropy() = default

Destructor.

void Init(const Reduction &reduction = MEAN)

Init. Refer to the parameters of Python API mindspore.ops.BinaryCrossEntropy for the inputs.

void set_reduction(const Reduction &reduction)

Set reduction.

Reduction get_reduction() const

Get reduction.

Returns

reduction.