Class NLLLoss

Defined in File nllloss.h

Inheritance Relationships

Base Type

public mindspore::ops::BaseOperator

Class Documentation

class NLLLoss : public mindspore::ops::BaseOperator 

NLLLoss operation. Refer to Python API mindspore.ops.NLLLoss for more details.

Public Functions

inline NLLLoss(): Constructor.

void Init(const Reduction &reduction = Reduction::NONE): Init. Refer to the parameters of Python API mindspore.ops.NLLLoss for the inputs.

void set_reduction(const Reduction &reduction): Set reduction.

Reduction get_reduction() const

Get reduction.

返回: reduction.