replay_trajectory_classification.environments.diffuse

diffuse(position_grid: np.ndarray, Fx: float, Fy: float, is_track_interior: np.ndarray, is_track_boundary: np.ndarray) → np.ndarray

Calculates diffusion for a single time step given a track.

Parameters:

• position_grid (np.ndarray, shape (n_bins_x, n_bins_y)) – Function to diffusion

• Fx (float) – Diffusion coefficient x

• Fy (float) – Diffusion coefficient y

• is_track_interior (np.ndarray, shape (n_bins_x, n_bins_y))

• is_track_boundary (np.ndarray, shape (n_bins_x, n_bins_y))

Returns:

diffused_grid

Return type:

np.ndarray, shape (n_bins_x, n_bins_y)