ucl-bug / jwave

A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs
GNU Lesser General Public License v3.0
135 stars 21 forks source link

Check the number of signal and number of sources positions #213

Open astanziola opened 11 months ago

astanziola commented 11 months ago

The user should receive an error when tries to do something like

sources = Sources(
  positions=((32,),(32,),(32,)),
  signals=jnp.stack([s1, s1]),
  dt=time_axis.dt,
  domain=domain,
)

since the number of signals is not the same as the number of sources positions