wrap¶
- class torchvision.tv_tensors.wrap(wrappee, *, like, **kwargs)[源代码]¶
将一个
torch.Tensor
(wrappee
) 转换为与like
相同的TVTensor
子类。如果
like
是一个BoundingBoxes
,则将like
的format
和canvas_size
分配给wrappee
,除非它们作为kwargs
传递。- 参数::
wrappee (Tensor) – 要转换的张量。
like (
TVTensor
) – 引用。wrappee
将转换为与like
相同的子类。kwargs – 如果
like
是一个BoundingBoxes
,则可以包含“format”和“canvas_size”。 否则将被忽略。
使用
wrap
的示例TVTensors 常见问题解答如何编写自己的 TVTensor 类