wrap

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

将一个torch.Tensor (wrappee) 转换为与 like 相同的 TVTensor 子类。

如果 like 是一个 BoundingBoxes，那么 like 的 format 和 canvas_size 将被分配给 wrappee，除非它们作为 kwargs 传递。

Parameters:

• wrappee (Tensor) – 要转换的张量。

• like (TVTensor) – 参考对象。 wrappee 将被转换为与 like 相同的子类。

• kwargs – 如果 like 是 BoundingBoxes，则可以包含“format”和“canvas_size”。 否则忽略。

使用 wrap 的示例：

TVTensors 常见问题

TVTensors FAQ

如何编写你自己的TVTensor类

How to write your own TVTensor class