espnet2.uasr.segmenter.random_segmenter.RandomSegmenter

source

class espnet2.uasr.segmenter.random_segmenter.RandomSegmenter(subsample_rate: float = 0.25, mean_pool: str2bool = True, mean_join_pool: str2bool = False, remove_zeros: str2bool = False)

Bases: AbsSegmenter

Initializes internal Module state, shared by both nn.Module and ScriptModule.

logit_segment(xs_pad: Tensor, padding_mask: Tensor) → Tensor

pre_segment(xs_pad: Tensor, padding_mask: Tensor) → Tensor