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维度必须为[0, 8]。

参数说明

返回值说明

返回一个ElementWiseLayer对象，如果添加失败则返回nullptr或抛出异常。