curobo.types.base module

class TensorDeviceType(

device: torch.device = device(type='cuda', index=0),

dtype: torch.dtype = torch.float32,

collision_geometry_dtype: torch.dtype = torch.float32,

collision_gradient_dtype: torch.dtype = torch.float32,

collision_distance_dtype: torch.dtype = torch.float32,

)

Bases: object

device: device = device(type='cuda', index=0)

dtype: dtype = torch.float32

collision_geometry_dtype: dtype = torch.float32

collision_gradient_dtype: dtype = torch.float32

collision_distance_dtype: dtype = torch.float32

static from_basic(

device: str,

dev_id: int,

)

to_device(data_tensor)

to_int8_device(

data_tensor,

)

cpu()

as_torch_dict()