cascade.model.CASCADE.jacobian

CASCADE.jacobian(adata, batch_size=128, n_devices=1)[source]

Compute the Jacobian matrix of the model

Parameters:

  • adata (AnnData) – Input dataset

  • batch_size (int) – Batch size

  • n_devices (int) – Number of GPU devices to use

Return type:

AnnData

Returns:

Dataset with –

  • layers["X_jac"]: The Jacobian matrix with shape (n_obs, n_vars, n_parents, n_particles)