trw.transforms.transforms_squeeze

Module Contents

Classes

TransformSqueeze

Squeeze a dimension of a tensor (i.e., remove one dimension of size 1 of a specifed axis)

Functions

squeeze_fn(feature_names: Sequence[str], batch: trw.basic_typing.Batch, axis: int) → trw.basic_typing.Batch

trw.transforms.transforms_squeeze.squeeze_fn(feature_names: Sequence[str], batch: trw.basic_typing.Batch, axis: int) → trw.basic_typing.Batch

class trw.transforms.transforms_squeeze.TransformSqueeze(axis: int, criteria_fn: Optional[trw.transforms.transforms.CriteriaFn] = criteria_is_array_4_or_above)

Bases: trw.transforms.transforms.TransformBatchWithCriteria

Squeeze a dimension of a tensor (i.e., remove one dimension of size 1 of a specifed axis)

Only numpy.ndarray and torch.Tensor types will be transformed