Coreml:如何在切片上添加两个张量？

import coremltools

# 创建一个切片层，提取第一个张量的切片
slice_layer1 = coremltools.proto.NeuralNetwork_pb2.SliceLayerParams()
slice_layer1.sliceAxis = 1  # 切片的轴
slice_layer1.startIndex = 0  # 切片的起始索引
slice_layer1.endIndex = 3  # 切片的结束索引

# 创建一个切片层，提取第二个张量的切片
slice_layer2 = coremltools.proto.NeuralNetwork_pb2.SliceLayerParams()
slice_layer2.sliceAxis = 1
slice_layer2.startIndex = 0
slice_layer2.endIndex = 3

# 创建一个相加层，将两个切片后的张量相加
add_layer = coremltools.proto.NeuralNetwork_pb2.AddLayerParams()

# 创建一个CoreML模型
model = coremltools.proto.Model_pb2.Model()
model.specificationVersion = 4

# 添加输入和输出
input_1 = model.description.input.add()
input_1.name = 'input_1'
input_1.type.multiArrayType.dataType = coremltools.proto.FeatureTypes_pb2.ArrayFeatureType.DOUBLE
input_1.type.multiArrayType.shape.extend([3])  # 输入张量的形状

input_2 = model.description.input.add()
input_2.name = 'input_2'
input_2.type.multiArrayType.dataType = coremltools.proto.FeatureTypes_pb2.ArrayFeatureType.DOUBLE
input_2.type.multiArrayType.shape.extend([3])

output = model.description.output.add()
output.name = 'output'
output.type.multiArrayType.dataType = coremltools.proto.FeatureTypes_pb2.ArrayFeatureType.DOUBLE
output.type.multiArrayType.shape.extend([3])

# 添加切片层和相加层到模型
model.description.layers.extend([slice_layer1, slice_layer2, add_layer])

# 保存模型
coremltools.utils.save_spec(model, 'model.mlmodel')