Batched Waypoint Suffix Sum Over 2D Trajectories

tesla

Given a 3D array traj with shape [batch, num_waypoint, 2], representing a 2D vector at each waypoint for each trajectory.

Compute a suffix-sum scalar sequence along the waypoint dimension. Fo...

Example

Unlock to view complete problem details

and practice with sample input/output

Was this article helpful?

View Test Cases & Run Code requires membership

Standard Input
Execution Result: