Function _gen_tile_idx

imperative/python/megengine/functional/loss.py:306–310 · view source on GitHub ↗

(inp: Tensor)

Source from the content-addressed store, hash-verified

304
305
306	def _gen_tile_idx(inp: Tensor):
307	idx = cumsum(inp, axis=0)
308	ret = ones(inp.sum(), dtype="int32")
309	ret[idx[:-1]] = -(inp - 1)[:-1]
310	return cumsum(ret, axis=0) - 1
311
312
313	def _expand_label(label: Tensor, label_lengths: Tensor, blank: int) -> Tensor:

_expand_labelFunction · 0.85

cumsumFunction · 0.85

onesFunction · 0.70

sumMethod · 0.45

no test coverage detected