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 的示例