ElementWiseLayer *AddElementWise(Tensor *input0, Tensor *input1, ElementWiseOperation op) noexcept;

函数功能

在网络中添加一个elementwise层。

函数原型

ElementWiseLayer *AddElementWise(Tensor *input0, Tensor *input1, ElementWiseOperation op) noexcept;

约束说明

• 输入的Tensor秩要保持一致。
• 输入的Tensor数据类型要保持一致。
• 输入的shape，对应维度必须相同，若不相同，则至少其中一个输入对应维度为1，支持广播。
• op必须是ElementWiseOperation已支持的类型。
• input0 , input1维度必须在[1, 8]。
• 当前不支持动态shape。

参数说明

返回值说明

返回一个ElementWiseLayer对象，如果添加失败时返回nullptr。