创建网络
- 创建builder。
import ascendie as aie builder = aie.Builder.create_builder(b'Ascend310P3')
- 使用builder创建network。
network = builder.create_network()
- 向Network中添加输入。
inputTensor0 = network.add_input(b"input0", aie.DataType.int32, aie.Dims([2,2])) # 添加输入tensor inputTensor1 = network.add_input(b"input1", aie.DataType.int32, aie.Dims([2,2]))
- 向Network中添加Layer。
addlayer = network.add_elementwise(inputTensor0, inputTensor1, aie.ElementWiseOperation.add) # 添加layer
- 向Network中添加输出。
network.mark_output(addlayer.get_output(0)) # 添加网络输出
父主题： Python编程模型