"""PyTorch custom op task."""