Struct State

Struct Documentation

struct State

Public Functions

State select(const std::vector<IndexType> &selIdx, int beamSize, bool isBatchMajor) const

Public Members

Expr output
Expr cell

Public Static Functions

static Expr select(Expr sel, const std::vector<IndexType> &selIdx, int beamSize, bool isBatchMajor)