def _tile_batch(self, t):
if t.shape.ndims is None or t.shape.ndims < 1:
raise ValueError("t must have statically known rank")
tiling = [1] * (t.shape.ndims + 1)
tiling[1] = self._beam_width
tiled = tf.tile(tf.expand_dims(t, 1), tiling)
return tiled
beam_aligner.py 文件源码
python
阅读 34
收藏 0
点赞 0
评论 0
评论列表
文章目录