已关闭
[Requirement|需求建议]: UnsortedSegmentMin支持空tensor #762
jinpenghe创建于  2月3日关闭于  2月6日
jinpenghe
jinpenghe成员
2月3日 创建

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;
}

likedislike
jinpenghe
jinpenghe成员
2月3日 评论:

/assign

likedislike
CANN-robotCANN-robot成员
2月3日 将 jinpenghe 设为负责人
jinpenghejinpenghe成员
2月3日 修改了issue 的描述
jinpenghejinpenghe成员
2月3日 修改了issue 的描述
CANN-robotCANN-robot成员
2月6日 关闭了 issue
CANN-robotCANN-robot成员
3月20日 添加了label:resolved