Class TopkGumbelSoftmaxStep

• Defined in File costs.h

Inheritance Relationships

Base Type

• public marian::models::ILogProbStep (Class ILogProbStep)

Class Documentation

class TopkGumbelSoftmaxStep : public marian::models::ILogProbStep

Public Functions

TopkGumbelSoftmaxStep(int k)

virtual ~TopkGumbelSoftmaxStep()

Ptr<DecoderState> apply(Ptr<DecoderState> state)