batchwise_max_force¶

torch_sim.quantities.batchwise_max_force(state)[source]¶

Compute the maximum force per batch.

Parameters:: state (SimState) – State to compute the maximum force per batch for.
Returns:: Maximum forces per batch
Return type:: Tensor