How to implement BN layers in TensorRT?

AI & Data Science Deep Learning (Training & Inference) TensorRT
1

Ubuntu 16.04 LTS
GPU type:1050Ti
nvidia driver version:390.87
CUDA version:9.0
CUDNN version:7.13
Python version:3.5
TensorRT version: 5.0.0.10

I have to create my network from scratch using the python API.It seems like that IScaleLayer could be used to achieve this goal.Any sample code or tutorial?
Thanks!

2

tf.layers.BatchNorm

3

@1006740126

Any detail or sample code?
I mean using TensorRT API from scartch to creat network,like the sample ‘network_api_pytorch_mnist’.
I don’t import from tensorflow with using uff.My model is pytorch one and couldn’t be convert to onnx format.