Hi,
Thanks for your answer.
I apologize for opening an issue both on PyTorch discuss and GitHub, I did not know which platform was the most adapted.
I will redirect this discussion to github issue 47575 14.