diff --git a/include/aidge/backend/cuda/data/TensorImpl.hpp b/include/aidge/backend/cuda/data/TensorImpl.hpp
index ea63f581400b5dc0669793a8d5f4c351885a9bb9..73b211113b3b21b9c8294e51e16cc001afad25e1 100644
--- a/include/aidge/backend/cuda/data/TensorImpl.hpp
+++ b/include/aidge/backend/cuda/data/TensorImpl.hpp
@@ -58,7 +58,11 @@ class TensorImpl_cuda : public TensorImpl, public TensorImpl_cuda_  {
     void *rawPtr() override {
         lazyInit(reinterpret_cast<void**>(&mData));
         return mData;
-    };
+    }
+
+    void* getRaw(std::size_t idx) {
+        return static_cast<void*>(static_cast<T*>(rawPtr()) + idx);
+    }
 
     const cudnnTensorDescriptor_t& getCudnnTensorDesc() const override {
         if (mCudnnTensor == nullptr) {