From 4c44d632dc651d87ad4cb9b54664d8b7ecaece52 Mon Sep 17 00:00:00 2001
From: akulov <aaakulov_3@edu.hse.ru>
Date: Thu, 9 Oct 2025 09:40:03 +0300
Subject: [PATCH] fix invalid tensors

---
 src/torchcodec/_core/Encoder.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/torchcodec/_core/Encoder.cpp b/src/torchcodec/_core/Encoder.cpp
index 9a1f4ee87..fce49f368 100644
--- a/src/torchcodec/_core/Encoder.cpp
+++ b/src/torchcodec/_core/Encoder.cpp
@@ -17,6 +17,11 @@ torch::Tensor validateSamples(const torch::Tensor& samples) {
       samples.dim() == 2,
       "samples must have 2 dimensions, got ",
       samples.dim());
+  TORCH_CHECK(
+      samples.numel() == 0 || samples.data_ptr() != nullptr,
+      "Samples tensor has null data pointer but non-zero numel. ",
+      "This indicates an invalid tensor state.");
+
 
   // We enforce this, but if we get user reports we should investigate whether
   // that's actually needed.