Class BinaryCrossEntropy

Defined in File binary_cross_entropy.h

Inheritance Relationships

Base Type

public mindspore::ops::BaseOperator

Class Documentation

class BinaryCrossEntropy : public mindspore::ops::BaseOperator 

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.

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.