Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
UnsortedSegmentMin支持空tensor。原代码在空tensor场景,如:[[4, 4, 3016, 0, 4, 7, 2], [4, 4, 3016, 0]] 没有做任何操作,导致输出错误。
NA
使得算子能够支持数据类型在空tensor场景下的正确输出。
空tensor场景走全载模板,调用init进行初始化为最大值后返回即可。
if (inputOuterDimSize * innerDimSize == 0) { InitGm<TX, Mode>(output, outputOuterDimSize * innerDimSize); return; }
/assign
Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
Backgroud(背景信息)
UnsortedSegmentMin支持空tensor。原代码在空tensor场景,如:[[4, 4, 3016, 0, 4, 7, 2], [4, 4, 3016, 0]] 没有做任何操作,导致输出错误。
Origin(信息来源)
NA
Benefit / Necessity (价值/作用)
使得算子能够支持数据类型在空tensor场景下的正确输出。
Design(设计方案)
空tensor场景走全载模板,调用init进行初始化为最大值后返回即可。
if (inputOuterDimSize * innerDimSize == 0) {
InitGm<TX, Mode>(output, outputOuterDimSize * innerDimSize);
return;
}