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 = 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.

Returns

reduction.