(label: Tensor, label_lengths: Tensor, blank: int)
| 311 | |
| 312 | |
| 313 | def _expand_label(label: Tensor, label_lengths: Tensor, blank: int) -> Tensor: |
| 314 | N = label_lengths.shape[0] |
| 315 | if len(label.shape) == 1: |
| 316 | L = label_lengths.max() |
| 317 | unpack_label = zeros((N, L), dtype="int32") + blank |
| 318 | idx_0 = _gen_repeat_idx(label_lengths) |
| 319 | idx_1 = _gen_tile_idx(label_lengths) |
| 320 | unpack_label[idx_0, idx_1] = label |
| 321 | label = unpack_label |
| 322 | |
| 323 | L = label.shape[1] |
| 324 | ex_label = zeros((N, L * 2 + 1), dtype="int32") + blank |
| 325 | ex_label[:, 1::2] = label |
| 326 | return ex_label |
| 327 | |
| 328 | |
| 329 | def _safelog(x: Tensor) -> Tensor: |
no test coverage detected