match PyTorch syntax
3 files changed