Class Logits

• Defined in File logits.h

Nested Relationships

Nested Types

• Struct Logits::MaskedFactorIndices

Class Documentation

class Logits

Public Functions

Logits()

Logits(Ptr<RationalLoss> logits)

Logits(Expr logits)

Logits(std::vector<Ptr<RationalLoss>> &&logits, Ptr<FactoredVocab> embeddingFactorMapping)

Expr getLogits() const

Expr getFactoredLogits(size_t groupIndex, Ptr<data::Shortlist> shortlist = nullptr, const std::vector<IndexType> &hypIndices = {}, size_t beamSize = 0) const

Expr applyLossFunction(const Words &labels, const std::function<Expr(Expr, Expr)> &lossFn) const

Logits applyUnaryFunction(const std::function<Expr(Expr)> &f) const

Logits applyUnaryFunctions(const std::function<Expr(Expr)> &f1, const std::function<Expr(Expr)> &fother) const

std::vector<Logits::MaskedFactorIndices> factorizeWords(const Words &words) const

Tensor getFactoredLogitsTensor(size_t factorGroup) const

size_t getNumFactorGroups() const

bool empty() const

Logits withCounts(const Expr &count) const

struct MaskedFactorIndices

Public Functions

void reserve(size_t n)

void push_back(size_t factorIndex)

MaskedFactorIndices()

MaskedFactorIndices(const Words &words)

Public Members

std::vector<WordIndex> indices

std::vector<float> masks