Whenever possible, we should try to raise exceptions through TORCH_CHECK instead of throw in our C++ code-base (follow-up to #535 (comment)).

This may require updates to the Python and C++ tests as well to catch the proper exception type.