kd.data.ValueRange

kd.data.ValueRange#

class kauldron.data.ValueRange(*, key: typing.Annotated[typing.Any, <object object at 0x7824c478ba80>] | typing.Sequence[typing.Annotated[typing.Any, <object object at 0x7824c478ba80>]] | dict[typing.Annotated[typing.Any, <object object at 0x7824c478ba80>], typing.Annotated[typing.Any, <object object at 0x7824c478ba80>]], 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]