kd.data.py

[[Source]]

PyGrain public API.

Symbols

Class

kd.data.py.AddConstants	Adds constant elements.
kd.data.py.Cast	Cast an element to the specified dtype.
kd.data.py.DataSource	Generic loader of arbitrary grain data source.
kd.data.py.DataSourceBase	Base class to implement a data source.
kd.data.py.ElementWiseTransform	Base class for elementwise transforms.
kd.data.py.Elements	Modify the elements by keeping xor dropping and/or renaming and/or copying.
kd.data.py.Gather	Gathers entries along a single dimension.
kd.data.py.HuggingFace	HuggingFace loader.
kd.data.py.Json	Json pipeline.
kd.data.py.Mix	Create a dataset mixture.
kd.data.py.PyGrainPipeline	Abstract base class to construct PyGrain data pipeline.
kd.data.py.Rearrange	Einops rearrange on a single element.
kd.data.py.Resize	Resizes an image.
kd.data.py.SliceDataset	Transform which select a subset of the dataset.
kd.data.py.Tfds	Base TFDS loader.
kd.data.py.TreeFlattenWithPath	Flatten any tree-structured elements.
kd.data.py.ValueRange	Map the value range of an element.