Hi all, Found this issue when trying to export small torch module that requires torch.autograd. Below works fine for scalar output case but does not for vector output. import torch #notice how .sum() ...

This code block includes necessary headers for CUDA programming with PyTorch. It also defines some macros and a utility function: CHECK_CUDA(x): A macro that checks if a given tensor x is a CUDA ...