wrap

class torchvision.tv_tensors.wrap(wrappee, *, like, **kwargs)

Convert a torch.Tensor (wrappee) into the same TVTensor subclass as like.

If like is a BoundingBoxes, the format and canvas_size of like are assigned to wrappee, unless they are passed as kwargs.

Parameters:

• wrappee (Tensor) – The tensor to convert.

• like (TVTensor) – The reference. wrappee will be converted into the same subclass as like.

• kwargs – Can contain “format” and “canvas_size” if like is a BoundingBoxes. Ignored otherwise.

Examples using wrap:

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