Toybrick
标题:
ReduceSum算子编译失败
[打印本页]
作者:
gef1998
时间:
4 天前
标题:
ReduceSum算子编译失败
onnx如下：
报错信息如下：
对应torch的代码为：
mask_mems
=
(
torch
.
abs
(
feat_mems
)
>
0
).
float
() #
feat_mems的shape为
[1, 6, 128, 50, 8, 50]
denom
=
torch
.
sum
(
mask_mems
,
dim
=
1
,
keepdim
=
False
)
欢迎光临 Toybrick (https://t.rock-chips.com/)
Powered by Discuz! X3.3
