FastFlowNet in TensorRT API

hello, I am trying to rewrite FastFlowNet in TensorRT API, and confused by some problems:
1, what’s diffference between IShuffleLayer::setFirstTranspose and IShuffleLayer::setSecondTranspose
2, how can I implememt the torch::index_select operation?
3, xx = torch.arange(0, W).view(1, -1).repeat(H, 1) ?
4, xx_plus = 2.0 * xx_plus / max(W-1,1) - 1.

fastFlowNet:

fastflownet.py (15.3 KB)

Hi,
We recommend you to check the below samples links in case of tf-trt integration issues.

If issue persist, We recommend you to reach out to Tensorflow forum.
Thanks!