视频¶
- class torchvision.tv_tensors.Video(data: Any, *, dtype: Optional[dtype] = None, device: Optional[Union[device, str, int]] = None, requires_grad: Optional[bool] = None)[source]¶
torch.Tensor
子类,用于形状为[..., T, C, H, W]
的视频。- Parameters:
data (tensor-like) – 任何可以通过
torch.as_tensor()
转换为张量的数据。dtype (torch.dpython:type, optional) – 期望的数据类型。如果省略,将从
data
中推断。device (torch.device, optional) – 期望的设备。如果省略且
data
是torch.Tensor
,则设备取自它。否则,视频将在 CPU 上构建。requires_grad (bool, 可选) – 是否应记录自动梯度操作。如果省略且
data
是torch.Tensor
,则从其取值。否则,默认为False
。
使用
Video
的示例: