TVTensors
TVTensors are torch.Tensor subclasses which the v2 transforms use under the hood to dispatch their inputs to the appropriate
lower-level kernels. Most users do not need to manipulate TVTensors directly.
Refer to
Getting started with transforms v2 for
an introduction to TVTensors, or
TVTensors FAQ for more advanced
info.
Image(data, *[, dtype, device, requires_grad])
|
torch.Tensor subclass for images.
|
Video(data, *[, dtype, device, requires_grad])
|
torch.Tensor subclass for videos.
|
BoundingBoxFormat(value)
|
Coordinate format of a bounding box. |
BoundingBoxes(data, *, format, canvas_size)
|
torch.Tensor subclass for bounding boxes.
|
Mask(data, *[, dtype, device, requires_grad])
|
torch.Tensor subclass for segmentation and detection masks.
|
TVTensor
|
Base class for all TVTensors. |
set_return_type(return_type)
|
Set the return type of torch operations on TVTensor. |
wrap(wrappee, *, like, **kwargs)
|
Convert a torch.Tensor (wrappee) into the same TVTensor subclass as like. |