From 8473ee6cd99ea17a1e25c2feeb3d368c23fc1673 Mon Sep 17 00:00:00 2001 From: Prabhat Roy Date: Tue, 1 Mar 2022 15:53:48 -0800 Subject: [PATCH] Remove unnecessary initialisation --- torchvision/csrc/io/decoder/gpu/gpu_decoder.cpp | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/torchvision/csrc/io/decoder/gpu/gpu_decoder.cpp b/torchvision/csrc/io/decoder/gpu/gpu_decoder.cpp index d26e80da2d8..1fe3ec8ab7a 100644 --- a/torchvision/csrc/io/decoder/gpu/gpu_decoder.cpp +++ b/torchvision/csrc/io/decoder/gpu/gpu_decoder.cpp @@ -29,8 +29,7 @@ torch::Tensor GPUDecoder::decode() { unsigned long videoBytes = 0; uint8_t* video = nullptr; at::cuda::CUDAGuard device_guard(device); - auto options = torch::TensorOptions().dtype(torch::kU8).device(torch::kCUDA); - torch::Tensor frame = torch::zeros({0}, options); + torch::Tensor frame; do { demuxer.demux(&video, &videoBytes); decoder.decode(video, videoBytes);