PyTorch Forums
Affine_grid_generator
C++
Cganzorig
(Cganzorig)
August 13, 2019, 12:45am
1
Where can I find torch.cudnn_affine_grid_generator(theta, N, C, H, W)? I need to modify it to 3d.