thank you Patrick. That was my observation too.
However I was caught by surprise since there seems to be open issues regarding comprehension syntaxes cf. More robust list comprehension · Issue #48153 · pytorch/pytorch · GitHub 15
But maybe that only relates to jit…