def add_elementwise(self: ascendie.Network, arg0: ascendie.Tensor, arg1: ascendie.Tensor, arg2: ascendie.ElementWiseOperation) -> ascendie.ElementWiseLayer

函数功能

在网络中添加一个elementwise层。

函数原型

add_elementwise(self: ascendie.Network, arg0: ascendie.Tensor, arg1: ascendie.Tensor, arg2: ascendie.ElementWiseOperation) -> ascendie.ElementWiseLayer

约束说明

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

参数说明