kd.data

[[Source]]

Data modules.

Symbols

Module

kd.data.iters	Dataset iterators.
kd.data.py	PyGrain public API.

Class

kd.data.AddConstants	Adds constant elements.
kd.data.BatchSize	Batch size.
kd.data.Cast	Cast an element to the specified dtype.
kd.data.CenterCrop	Crop the input data to the specified shape from the center.
kd.data.ElementWiseTransform	Base class for elementwise transforms.
kd.data.Elements	Modify the elements by keeping xor dropping and/or renaming and/or copying.
kd.data.FilterTransform	Abstract base class for filter transformations for individual elements.
kd.data.Gather	Gathers entries along a single dimension.
kd.data.InMemoryPipeline	Pipeline which fit in memory.
kd.data.IterableDataset	General interface for iterable datasets.
kd.data.MapTransform	Abstract base class for all 1:1 transformations of elements.
kd.data.Pipeline	Base class for kauldron data pipelines.
kd.data.Rearrange	Einops rearrange on a single element.
kd.data.Resize	Resizes an image.
kd.data.TreeFlattenWithPath	Flatten any tree-structured elements.
kd.data.ValueRange	Map the value range of an element.