Class SparseSoftmaxCrossEntropyWithLogits

Defined in File sparse_softmax_cross_entropy_with_logits.h

Inheritance Relationships

Base Type

public mindspore::ops::BaseOperator

Class Documentation

class SparseSoftmaxCrossEntropyWithLogits : public mindspore::ops::BaseOperator 

Computes the softmax cross-entropy value between logits and sparse encoding labels. Refer to Python API mindspore.ops.SparseSoftmaxCrossEntropyWithLogits for more details.

Public Functions

inline SparseSoftmaxCrossEntropyWithLogits(): Constructor.

void Init(const bool is_grad = false): Init. Refer to the parameters of python API mindspore.ops.SparseSoftmaxCrossEntropyWithLogits for the inputs.

void set_is_grad(const bool is_grad): Set is_grad.

bool get_is_grad() const

Get is_grad.

Returns: is_grad.