def ctc_decode(y): y = y.argmax(-1) t = y.size(0) result = [] for i in range(t+1): result.append(MyDataset.ctc_arr2txt(y[:i], start=1)) return result
def ctc_decode(y): result = [] y = y.argmax(-1) return [MyDataset.ctc_arr2txt(y[_], start=1) for _ in range(y.size(0))]
def ctc_decode(y): y = y.argmax(-1) return [MyDataset.ctc_arr2txt(y[_]) for _ in range(y.size(0))]