curobo.util.tensor_util module

check_tensor_shapes( new_tensor: Tensor, mem_tensor: Tensor, )

copy_tensor( new_tensor: Tensor, mem_tensor: Tensor, )

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_

clone_if_not_none(x)

Clones x if it’s not None.

Parameters:: x (torch.Tensor) – _description_
Returns:: _description_
Return type:: _type_

tensor_repeat_seeds(tensor, num_seeds)