## <!-- Name of operator to add support for --> - **Function Schema**: - **Original PyTorch API**: - **Relevant TensorRT Documentation**: ## Alternatives <!-- A clear and concise description of any alternative solutions you've considered, if any. --> ## Additional context <!-- Add any other context or screenshots about the operator request here. -->