Method position_encoding

OSUM/wenet/transformer/subsampling.py:32–34

(self, offset: Union[int, torch.Tensor],
                          size: int)

        self.subsampling_rate = 1

    def position_encoding(self, offset: Union[int, torch.Tensor],
                          size: int) -> torch.Tensor:
        return self.pos_enc.position_encoding(offset, size)


class EmbedinigNoSubsampling(BaseSubsampling):