标签： tensorflow tensorrt
我想为Tensorflow模型加速使用Tensorrt的推理。 首先，我将PB模型转换为uff模型， 然后我解析uff模型，但出现以下错误：
ERROR: UFFParser: Validator error: ExpandDims_36: Unsupported operation _ExpandDims ERROR: sample_uff_mnist: Fail to parse
似乎tensorrt不支持tf.expand_dims操作。 我该怎么做才能解决这个问题？