
例如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.
例如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.