oneflow
oneflow copied to clipboard
ctc_loss support long input
背景:https://github.com/Oneflow-Inc/OneCloud/issues/104#issuecomment-1156122806 概述:oneflow的ctc_loss不接受int64的target输入而torch能 实现:
- ctc_loss共有四个输入参数,除log_prob之外,其余都是整数形式。实现过程中发现torch支持三个整数参数 int32/64 任意类型的组合,而oneflow只能同时为int64或int32。故增加了两个模板参数,和torch的接口对齐。
文档: