请问构建算子时,None这种类型要怎么表示?
收藏回复举报
请问构建算子时,None这种类型要怎么表示?
t('forum.solved') 已解决
发表于2023-04-07 10:40:26
0 查看

cke_184.png

例如reduce_mean 这个算子,axes 如果是none的话,会reduce所有维度,那么我要出入 none呢?

    auto reduce_mean = op::ReduceMean("reduce_mean")

        .set_input_x(view)

        .set_attr_axes()    // 这里要填什么?

        .set_attr_keep_dims(false);

    graph_outputs.push_back(reduce_mean);

如果 set_attr_axes()  不设置的话,运行时候会报错

[ERROR] GE(51962,python):2023-04-07-02:38:24.398.221 [node.cc:717]51962 Verify: ErrorNo: 4294967295(failed) [COMP][PRE_OPT][Check][Param] operator reduce_mean's input 1 is not linked.

我要发帖子