Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
DeepSpeech
提交
1cdd41bd
D
DeepSpeech
项目概览
PaddlePaddle
/
DeepSpeech
大约 2 年 前同步成功
通知
210
Star
8425
Fork
1598
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
245
列表
看板
标记
里程碑
合并请求
3
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
D
DeepSpeech
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
245
Issue
245
列表
看板
标记
里程碑
合并请求
3
合并请求
3
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
1cdd41bd
编写于
5月 24, 2022
作者:
H
huangyuxin
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix pad_sequence, test=asr
上级
ea71fddb
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
3 addition
and
3 deletion
+3
-3
paddlespeech/s2t/utils/tensor_utils.py
paddlespeech/s2t/utils/tensor_utils.py
+3
-3
未找到文件。
paddlespeech/s2t/utils/tensor_utils.py
浏览文件 @
1cdd41bd
...
@@ -82,7 +82,7 @@ def pad_sequence(sequences: List[paddle.Tensor],
...
@@ -82,7 +82,7 @@ def pad_sequence(sequences: List[paddle.Tensor],
max_size
=
sequences
[
0
].
size
()
max_size
=
sequences
[
0
].
size
()
# (TODO Hui Zhang): slice not supprot `end==start`
# (TODO Hui Zhang): slice not supprot `end==start`
# trailing_dims = max_size[1:]
# trailing_dims = max_size[1:]
trailing_dims
=
max_size
[
1
:]
if
max_size
.
ndim
>=
2
else
()
trailing_dims
=
tuple
(
max_size
[
1
:].
numpy
().
tolist
())
if
sequences
[
0
]
.
ndim
>=
2
else
()
max_len
=
max
([
s
.
shape
[
0
]
for
s
in
sequences
])
max_len
=
max
([
s
.
shape
[
0
]
for
s
in
sequences
])
if
batch_first
:
if
batch_first
:
out_dims
=
(
len
(
sequences
),
max_len
)
+
trailing_dims
out_dims
=
(
len
(
sequences
),
max_len
)
+
trailing_dims
...
@@ -99,7 +99,7 @@ def pad_sequence(sequences: List[paddle.Tensor],
...
@@ -99,7 +99,7 @@ def pad_sequence(sequences: List[paddle.Tensor],
if
batch_first
:
if
batch_first
:
# TODO (Hui Zhang): set_value op not supprot `end==start`
# TODO (Hui Zhang): set_value op not supprot `end==start`
# TODO (Hui Zhang): set_value op not support int16
# TODO (Hui Zhang): set_value op not support int16
# TODO (Hui Zhang): set_varbase 2 rank not support [0,0,...]
# TODO (Hui Zhang): set_varbase 2 rank not support [0,0,...]
# out_tensor[i, :length, ...] = tensor
# out_tensor[i, :length, ...] = tensor
if
length
!=
0
:
if
length
!=
0
:
out_tensor
[
i
,
:
length
]
=
tensor
out_tensor
[
i
,
:
length
]
=
tensor
...
@@ -145,7 +145,7 @@ def add_sos_eos(ys_pad: paddle.Tensor, sos: int, eos: int,
...
@@ -145,7 +145,7 @@ def add_sos_eos(ys_pad: paddle.Tensor, sos: int, eos: int,
[ 4, 5, 6, 11, -1, -1],
[ 4, 5, 6, 11, -1, -1],
[ 7, 8, 9, 11, -1, -1]])
[ 7, 8, 9, 11, -1, -1]])
"""
"""
# TODO(Hui Zhang): using comment code,
# TODO(Hui Zhang): using comment code,
#_sos = paddle.to_tensor(
#_sos = paddle.to_tensor(
# [sos], dtype=paddle.long, stop_gradient=True, place=ys_pad.place)
# [sos], dtype=paddle.long, stop_gradient=True, place=ys_pad.place)
#_eos = paddle.to_tensor(
#_eos = paddle.to_tensor(
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录