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

如何編寫自己的 v2 變換

如何編寫自己的 v2 變換

如何編寫自己的 TVTensor 類

如何編寫自己的 TVTensor 類

TVTensors 常見問題

TVTensors 常見問題