tenorrt не поддерживает tf.expand_dims: неподдерживаемая операция _ExpandDims
Я хочу ускорить вывод с помощью tenorrt для тензорной модели. Сначала я конвертирую модель pb в модель uff, затем я анализирую модель uff, но получаю следующую ошибку:
ERROR: UFFParser: Validator error: ExpandDims_36: Unsupported operation _ExpandDims
ERROR: sample_uff_mnist: Fail to parse
Кажется, тензор не поддерживает операцию tf.expand_dims. Что я должен сделать, чтобы решить эту проблему?