kd.data.py.ElementWiseRandomTransform

kd.data.py.ElementWiseRandomTransform#

class kauldron.data.py.ElementWiseRandomTransform(*, key: typing.Annotated[typing.Any, <object object at 0x76412092fb90>] | typing.Sequence[typing.Annotated[typing.Any, <object object at 0x76412092fb90>]] | dict[typing.Annotated[typing.Any, <object object at 0x76412092fb90>], typing.Annotated[typing.Any, <object object at 0x76412092fb90>]])[source]

Bases: kauldron.data.transforms.base.ElementWise, grain._src.core.transforms.RandomMapTransform

Base class for elementwise transforms.

random_map(
features,
rng: numpy.random._generator.Generator,
)[source]

Maps a single element.

abstractmethod random_map_element(
element: Any,
rng: numpy.random._generator.Generator,
)[source]