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

如何編寫您自己的 v2 轉換

如何編寫您自己的 v2 轉換

TVTensors 常見問題解答

TVTensors 常見問題解答

如何編寫您自己的 TVTensor 類別

如何編寫您自己的 TVTensor 類別