torch_utils#

Utility functions for PyTorch in ClearMap.

gpu_info()[source]#

set_type(torch_obj, dtype, cuda)[source]#

to(t, dtype=<class 'float'>)[source]#

Convert torch object to a specified data type.

Arguments

ttorch object: The object to convert to a crtian data type.
dtype[‘float’, ‘double’, ‘float64’, ‘float32’, float16’, ‘half’, float]: The data type to use for the torch object.

Returns

ttorch object: The torch object in the requested data type.