curobo.curobolib.tensor_step_cu module

step_acceleration(

arg0: torch.Tensor,

arg1: torch.Tensor,

arg2: torch.Tensor,

arg3: torch.Tensor,

arg4: torch.Tensor,

arg5: torch.Tensor,

arg6: torch.Tensor,

arg7: torch.Tensor,

arg8: torch.Tensor,

arg9: int,

arg10: int,

arg11: int,

arg12: bool,

) → list[torch.Tensor]

Tensor Step Acceleration (curobolib)

step_acceleration_idx(

arg0: torch.Tensor,

arg1: torch.Tensor,

arg2: torch.Tensor,

arg3: torch.Tensor,

arg4: torch.Tensor,

arg5: torch.Tensor,

arg6: torch.Tensor,

arg7: torch.Tensor,

arg8: torch.Tensor,

arg9: torch.Tensor,

arg10: int,

arg11: int,

arg12: int,

arg13: bool,

) → list[torch.Tensor]

Tensor Step Acceleration (curobolib)

step_idx_position2(

arg0: torch.Tensor,

arg1: torch.Tensor,

arg2: torch.Tensor,

arg3: torch.Tensor,

arg4: torch.Tensor,

arg5: torch.Tensor,

arg6: torch.Tensor,

arg7: torch.Tensor,

arg8: torch.Tensor,

arg9: torch.Tensor,

arg10: int,

arg11: int,

arg12: int,

arg13: int,

) → list[torch.Tensor]

Tensor Step Position (curobolib)

step_position(

arg0: torch.Tensor,

arg1: torch.Tensor,

arg2: torch.Tensor,

arg3: torch.Tensor,

arg4: torch.Tensor,

arg5: torch.Tensor,

arg6: torch.Tensor,

arg7: torch.Tensor,

arg8: torch.Tensor,

arg9: int,

arg10: int,

arg11: int,

) → list[torch.Tensor]

Tensor Step Position (curobolib)

step_position2(

arg0: torch.Tensor,

arg1: torch.Tensor,

arg2: torch.Tensor,

arg3: torch.Tensor,

arg4: torch.Tensor,

arg5: torch.Tensor,

arg6: torch.Tensor,

arg7: torch.Tensor,

arg8: torch.Tensor,

arg9: int,

arg10: int,

arg11: int,

arg12: int,

) → list[torch.Tensor]

Tensor Step Position (curobolib)

step_position_backward(

arg0: torch.Tensor,

arg1: torch.Tensor,

arg2: torch.Tensor,

arg3: torch.Tensor,

arg4: torch.Tensor,

arg5: torch.Tensor,

arg6: int,

arg7: int,

arg8: int,

) → list[torch.Tensor]

Tensor Step Position (curobolib)

step_position_backward2(

arg0: torch.Tensor,

arg1: torch.Tensor,

arg2: torch.Tensor,

arg3: torch.Tensor,

arg4: torch.Tensor,

arg5: torch.Tensor,

arg6: int,

arg7: int,

arg8: int,

arg9: int,

) → list[torch.Tensor]

Tensor Step Position (curobolib)