mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-10 18:42:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

consider case of empty tensor (#540)

Browse Source
This commit is contained in:
Zengwei Yao 2022-08-22 21:42:56 +08:00 committed by GitHub
parent 9277c95bcd
commit c0101185d7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
icefall/diagnostics.py
View File

@ -130,6 +130,8 @@ class TensorDiagnostic(object):
x = x[0] x = x[0]
if not isinstance(x, Tensor): if not isinstance(x, Tensor):
return return
if x.numel() == 0: # for empty tensor
return
x = x.detach().clone() x = x.detach().clone()
if x.ndim == 0: if x.ndim == 0:
x = x.unsqueeze(0) x = x.unsqueeze(0)