Torch

Utility functions for PyTorch in ClearMap.

gpu_info()

to(t, dtype=<class 'float'>)

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.