快捷方式

wrap

class torchvision.tv_tensors.wrap(wrappee, *, like, **kwargs)[源代码]

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

如果 like 是一个 BoundingBoxes,则将 likeformatcanvas_size 分配给 wrappee,除非它们作为 kwargs 传递。

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

  • like (TVTensor) – 引用。 wrappee 将转换为与 like 相同的子类。

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

使用 wrap 的示例

TVTensors 常见问题解答

TVTensors 常见问题解答

如何编写自己的 TVTensor 类

如何编写自己的 TVTensor 类

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题解答

查看资源