batch_matrix_vector¶
- torch_sim.integrators.npt.batch_matrix_vector(matrices, vectors)[source]¶
Perform batch matrix-vector multiplication.
- Parameters:
matrices (
torch.Tensor) – shape [n_systems, n, n]vectors (
torch.Tensor) – shape [n_systems, n, m]
- Returns:
shape [n_systems, n, m] result of multiplication
- Return type: