TensorRT Python API拼接层定义报错：ITensor*转ITensor*const*咨询

我来帮你搞定这个拼接层的错误问题，你遇到的TypeError主要是因为对API参数的理解和用法有误，结合SWIG封装的特性，咱们一步步修正：

错误原因分析 #

你写的代码concatLayer = network.add_concatenation([conv1.get_output(0),conv2.get_output(0)],2)有两个核心问题：

1. 参数顺序错误：根据你提供的C++ API文档，addConcatenation的第二个参数是输入张量的数量nbInputs，而不是拼接维度。你错误地把拼接维度2放在了这个位置，导致类型不匹配（函数期望接收张量指针数组，你却传了整数）。
2. SWIG封装的参数适配：虽然conv1.get_output(0)返回的是ITensor*类型，但直接传递Python列表在旧版本SWIG封装中需要适配，不过新版本TensorRT已经简化了这个流程。

正确的代码写法 #

新版本TensorRT的Python API已经不需要手动传入nbInputs（SWIG会自动从输入列表推导数量），你只需要先创建拼接层，再单独设置拼接维度：

# 传递输入张量列表，创建拼接层
concat_layer = network.add_concatenation([conv1.get_output(0), conv2.get_output(0)])
# 设置拼接的维度（这里你要的是维度2）
concat_layer.set_axis(2)

如果你的TensorRT版本比较旧，确实需要手动传入nbInputs，可以这样写（确保第二个参数是输入数量，而非维度）：

inputs = [conv1.get_output(0), conv2.get_output(0)]
# 第二个参数是输入张量的数量，这里是2
concat_layer = network.add_concatenation(inputs, len(inputs))
concat_layer.set_axis(2)

重要注意事项 #

别忘了API文档里的警告：除了你要拼接的通道维度外，所有输入张量的其他维度必须完全相同，否则拼接层会创建失败。

内容的提问来源于stack exchange，提问作者WifiSpy