curobo.util.tensor_util module

curobo.util.tensor_util.check_tensor_shapes(new_tensor, mem_tensor)
Parameters:
  • new_tensor (Tensor) –

  • mem_tensor (Tensor) –

curobo.util.tensor_util.copy_tensor(new_tensor, mem_tensor)
Parameters:
  • new_tensor (Tensor) –

  • mem_tensor (Tensor) –

curobo.util.tensor_util.copy_if_not_none(new_tensor, ref_tensor)

Clones x if it’s not None. TODO: Rename this to clone_if_not_none

Parameters:

x (torch.Tensor) – _description_

Returns:

_description_

Return type:

_type_

curobo.util.tensor_util.clone_if_not_none(x)

Clones x if it’s not None.

Parameters:

x (torch.Tensor) – _description_

Returns:

_description_

Return type:

_type_

curobo.util.tensor_util.tensor_repeat_seeds(tensor, num_seeds)