Do we have a set_default_dtype in libtorch

7starsea · July 23, 2019, 3:43am

In python, we have torch.set_default_dtype, but I did not find a similar method in libtorch, do we have one?

Thanks.

dfalbel · June 25, 2020, 5:44pm

It’s defined here: https://github.com/pytorch/pytorch/blob/master/c10/core/DefaultDtype.h#L10
And can be accessed trough torch::set_default_dtype.