wrap

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

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

如果 like 是 BoundingBoxes，则除非通过 kwargs 传递，否则 like 的 format 和 canvas_size 将被赋给 wrappee。

参数：

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

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

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

使用 wrap 的示例

如何编写自己的 v2 变换

如何编写自己的 v2 变换

如何编写自己的 TVTensor 类

如何编写自己的 TVTensor 类

TVTensors 常见问题

TVTensors 常见问题