For more examples, see https://github.com/Xilinx/Vitis-AI/tree/2.0/tools/Vitis-AI-Optimizer/vai_p_pytorch.