zea.tensor\_ops =============== .. automodule:: zea.tensor_ops .. rubric:: Functions .. autosummary:: L2 add_salt_and_pepper_noise apply_along_axis batch_cov batched_map boolean_mask check_patches_fit compute_required_patch_overlap compute_required_patch_shape correlate extend_n_dims flatten fori_loop func_with_one_batch_dim gaussian_filter gaussian_filter1d images_to_patches interpolate_data is_jax_prng_key is_monotonic linear_sum_assignment map_indices_for_interpolation matrix_power nonzero pad_array_to_divisible patched_map patches_to_images resample reshape_axis safe_vectorize sinc split_seed split_volume_data_from_axis stack_volume_data_along_axis