TVTensor

class torchvision.tv_tensors.TVTensor

Base class for all TVTensors.

You probably don’t want to use this class unless you’re defining your own custom TVTensors. See How to write your own TVTensor class for details.

Examples using TVTensor:

Getting started with transforms v2

Getting started with transforms v2

Transforms v2: End-to-end object detection/segmentation example

Transforms v2: End-to-end object detection/segmentation example

How to write your own v2 transforms

How to write your own v2 transforms

TVTensors FAQ

TVTensors FAQ

How to write your own TVTensor class

How to write your own TVTensor class

property device: device

Is the torch.device where this Tensor is.

property ndim: int

Alias for dim()

property shape: Union[Size, List[int], Tuple[int, ...]]

Returns the size of the self tensor. Alias for size.

See also Tensor.size().

Example:

>>> t = torch.empty(3, 4, 5)
>>> t.size()
torch.Size([3, 4, 5])
>>> t.shape
torch.Size([3, 4, 5])