athena.alignment

Module Contents

Functions

insert_blank(label[, blank_id])	Insert blank token between every two label token.
force_align(probs, y[, blank_id])	Force alignment
alignment(jsonfile, config[, rank_size, rank])	CTC Alignmnet according encoder probs

Attributes

stream

athena.alignment.insert_blank(label, blank_id=0)

Insert blank token between every two label token.

athena.alignment.force_align(probs, y, blank_id=0)

Force alignment :param ctc_probs: hidden state sequence, 3d tensor (B,T, D) :param y: id sequence tensor 2d tensor (B,L) :param int blank_id: blank symbol index

Returns

alignment result (T)

athena.alignment.alignment(jsonfile, config, rank_size=1, rank=0)

CTC Alignmnet according encoder probs

athena.alignment.stream