desy-ml / cheetah

Fast and differentiable particle accelerator optics simulation for reinforcement learning and optimisation applications.
https://cheetah-accelerator.readthedocs.io
GNU General Public License v3.0
35 stars 14 forks source link

Fix vectorized tracking for `TransverseDeflectingCavity` #278

Closed cr-xu closed 1 month ago

cr-xu commented 1 month ago

Description

Adds some missing dimensional expansions in the TransverseDeflectingCavity code to fix a bug with some vectorisation setups.

Motivation and Context

Fixes #270.

Types of changes

Checklist

Note: We are using a maximum length of 88 characters per line.