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”和“clamping_mode”。否則會被忽略。

使用 wrap 的示例

如何編寫自己的 v2 變換

如何編寫自己的 v2 變換

如何編寫自己的 TVTensor 類

如何編寫自己的 TVTensor 類

TVTensor FAQ

TVTensor FAQ