快捷方式

wrap

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

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

如果 likeBoundingBoxes,则除非通过 kwargs 传递,否则 likeformatcanvas_size 将被赋给 wrappee

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

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

  • kwargs – 如果 likeBoundingBoxes,则可以包含 “format” 和 “canvas_size”。否则将被忽略。

使用 wrap 的示例

如何编写自己的 v2 变换

如何编写自己的 v2 变换

如何编写自己的 TVTensor 类

如何编写自己的 TVTensor 类

TVTensors 常见问题

TVTensors 常见问题

© Copyright 2017-至今,Torch 贡献者。

使用 Sphinx 构建,主题由 Read the Docs 提供。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源