pbrs_simd

Function pbrs_simd 

Source
pub fn pbrs_simd(
    rewards: &[f64],
    phi_current: &[f64],
    phi_next: &[f64],
    gamma: f64,
    dones: &[f64],
) -> Vec<f64>
Expand description

SIMD-friendly PBRS: result[i] = rewards[i] + gamma * phi_next[i] - phi_current[i]

At episode boundaries (dones[i] == 1.0), returns the raw reward. Processes 4 f64s at a time (256-bit AVX2 = 4 x f64).

ยงPanics

Panics if all slices are not the same length.