flow.stream

flow.stream(f)

Shift each population one cell along its own velocity (periodic).

Parameters

Name Type Description Default
f jax.Array Post-collision populations, (19, NX, NY, NZ). required

Returns

Name Type Description
jax.Array Streamed populations, (19, NX, NY, NZ).