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_

cat_sum( tensor_list: List[Tensor], )¶

cat_sum_horizon( tensor_list: List[Tensor], )¶

cat_max( tensor_list: List[Tensor], )¶

tensor_repeat_seeds(tensor, num_seeds)¶

fd_tensor( p: Tensor, dt: Tensor, )¶