runs on the GPU, and made affine_transform() take only tensors.
Attach a file by drag & drop or click to upload