Class CrossEntropyNodeOp

• Defined in File node_operators_binary.h

Inheritance Relationships

Base Type

• public marian::NaryNodeOp (Struct NaryNodeOp)

Class Documentation

class CrossEntropyNodeOp : public marian::NaryNodeOp

Public Functions

CrossEntropyNodeOp(Expr a, Expr indices, float labelSmoothingAlpha, Type outputType = Type::float32)

Shape newShape(Expr a)

NodeOps forwardOps()

NodeOps backwardOps()

virtual size_t hash()

virtual bool equal(Expr node)

const std::string type()