Thanks for the response. I found the new_* functions on Tensor that seem to provide similar functionality (though maybe more verbose). Not sure if a new feature is worth adding given that there’s an alternative. I suppose a context manager would shrink the API surface and make code more consistent…