kd.data.py.ValueRange#

class kauldron.data.py.ValueRange(*, 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>]], vrange: tuple[float, float], in_vrange: tuple[float, float] = (0.0, 255.0), dtype: typing.Any = <class 'numpy.float32'>, clip_values: bool = True)[source]

Bases: kauldron.data.transforms.base.ElementWiseTransform

Map the value range of an element.

vrange: tuple[float, float]

in_vrange: tuple[float, float] = (0.0, 255.0)

dtype: alias of numpy.float32

clip_values: bool = True

map_element( element: jaxtyping.Shaped[Tensor, '*any'] | jaxtyping.Shaped[ndarray, '*any'] | jaxtyping.Shaped[Array, '*any'], ) → jaxtyping.Shaped[Tensor, '*any'] | jaxtyping.Shaped[ndarray, '*any'] | jaxtyping.Shaped[Array, '*any'][source]

key: kontext.Key | Sequence[kontext.Key] | dict[kontext.Key, kontext.Key]