kd.data

Contents

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.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.