Clipping#

class vision_architectures.transforms.clipping.Clip(min_value, max_value)[source]#

Bases: Transform

__init__(min_value, max_value)[source]#
__call__(data)[source]#

Clip the input tensor to the specified range.

Parameters:

data (Tensor) – Input tensor to be clipped.

Return type:

Tensor

Returns:

Clipped tensor.

class vision_architectures.transforms.clipping.Clipd(keys, min_value, max_value)[source]#

Bases: Transform

__init__(keys, min_value, max_value)[source]#
__call__(data)[source]#

Clip the input tensor to the specified range.

Parameters:

data (dict[Hashable, Tensor]) – Input tensor to be clipped.

Return type:

dict[Hashable, Tensor]

Returns:

Clipped tensor.