WholeAxis

class of dascore.lazy.contract inherits from: ChunkContract

source

WholeAxis(
    axes: tuple[int, …] ,
)-> None

An output chunk needs the whole of the input along some axes.

Along every other axis it needs the input at the same place, so the work can still be split along those.

Parameters

Parameter	Description
axes	The axes the kernel needs whole.

Examples

from dascore.lazy.contract import WholeAxis
box = (slice(10, 20), slice(2, 3))
assert WholeAxis((0,)).needs(box, (100, 5)) == (slice(0, 100), slice(2, 3))

Methods

Name	Description
apply	Return the output box, computed from the input it needs.
needs	Return the box with each of the axes whole.