提交 6a9daa80 编写于 作者: H Hui Zhang

encoder.forward_one_step for prefix feat or ys

上级 0cd30e48
...@@ -383,7 +383,7 @@ class TransformerEncoder(BaseEncoder): ...@@ -383,7 +383,7 @@ class TransformerEncoder(BaseEncoder):
"""Encode input frame. """Encode input frame.
Args: Args:
xs (paddle.Tensor): Input tensor. (B, T, D) xs (paddle.Tensor): (Prefix) Input tensor. (B, T, D)
masks (paddle.Tensor): Mask tensor. (B, T, T) masks (paddle.Tensor): Mask tensor. (B, T, T)
cache (List[paddle.Tensor]): List of cache tensors. cache (List[paddle.Tensor]): List of cache tensors.
...@@ -396,7 +396,6 @@ class TransformerEncoder(BaseEncoder): ...@@ -396,7 +396,6 @@ class TransformerEncoder(BaseEncoder):
xs = self.global_cmvn(xs) xs = self.global_cmvn(xs)
if isinstance(self.embed, Conv2dSubsampling): if isinstance(self.embed, Conv2dSubsampling):
# xs, masks = self.embed(xs, masks)
#TODO(Hui Zhang): self.embed(xs, masks, offset=0), stride_slice not support bool tensor #TODO(Hui Zhang): self.embed(xs, masks, offset=0), stride_slice not support bool tensor
xs, pos_emb, masks = self.embed(xs, masks.astype(xs.dtype), offset=0) xs, pos_emb, masks = self.embed(xs, masks.astype(xs.dtype), offset=0)
else: else:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册