Class NLLLoss

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, const int64_t ignore_index = -100)

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.

void set_ignore_index(const int64_t ignore_index)

Set ignore_index.

int64_t get_ignore_index() const

Get ignore_index.

Returns

ignore_index.