Others¶
Lambda
¶
- class torchio.transforms.Lambda(function: Callable[[Tensor], Tensor], types_to_apply: Sequence[str] | None = None, **kwargs)[source]¶
Bases:
Transform
Applies a user-defined function as transform.
- Parameters:
function – Callable that receives and returns a 4D
torch.Tensor
.types_to_apply – List of strings corresponding to the image types to which this transform should be applied. If
None
, the transform will be applied to all images in the subject.**kwargs – See
Transform
for additional keyword arguments.
Example
>>> import torchio as tio >>> invert_intensity = tio.Lambda(lambda x: -x, types_to_apply=[tio.INTENSITY]) >>> invert_mask = tio.Lambda(lambda x: 1 - x, types_to_apply=[tio.LABEL]) >>> def double(x): ... return 2 * x >>> double_transform = tio.Lambda(double)