Module spec 

Observation specification types.

An ObsSpec defines how to extract flat observation tensors from simulation state. Each ObsEntry targets one field, specifying the spatial region to observe, the transform to apply, and the output data type.

Structs

ObsEntry

A single observation entry targeting one field over a spatial region.

ObsSpec

Specification for observation extraction.

PoolConfig

Configuration for spatial pooling applied after gather.

Enums

ObsDtype

Output data type for observation values.

ObsRegion

Observation region — how to select spatial cells for an entry.

ObsTransform

Transform applied to raw field values before output.

PoolKernel

Pooling kernel type for spatial downsampling.