Python Tensorflow tf.abs（）可转换为TensorRT的替代项
我正在尝试将Tensorflow冻结模型转换为NVIDIA TensorRT引擎。我有TensorRT 3版本。TensorRT不支持某些Tensorflow层/函数，如Python Tensorflow tf.abs（）可转换为TensorRT的替代项,python,tensorflow,nvidia,tensor,tensorrt,Python,Tensorflow,Nvidia,Tensor,Tensorrt,我正在尝试将Tensorflow冻结模型转换为NVIDIA TensorRT引擎。我有TensorRT 3版本。TensorRT不支持某些Tensorflow层/函数，如tf.Cast等。 我的模型使用tf.abs（）函数（编辑：实际上TensorRT支持该函数，但构建TensorRT引擎所需的UFF解析器不支持该函数），并且我能够在将tensorflow检查点转换为冻结模型之前更改操作 我需要用替代品替换tf.abs（）。因此，我尝试在Tensorflow中执行以下numpy操作 努比： ma
tf.Casttf.abs（）tf.abs（）mask = (x<0).astype(np.float32)
mask = 2*mask + 1 #(makes all 1s -1, and all 0s 1)
abs_vals = mask*x
tf.Castabs不幸的是，截至2018年4月，UFF转换不支持很多操作。这一个有效，相当于tf.abs（）：
对不起，我的错。TensorRT不支持
abstf.sqrttf.max（）mask = tf.less(x, z)
#convert mask from boolean to integer or float
abs_val = tf.maximum( 0, x ) + tf.maximum( 0, -x )