From 8a2e68a4e9b0e66922a0166afc946472d185d5dd Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Wed, 12 Jun 2024 09:40:29 +0200
Subject: [PATCH] Implement Tensor::zeros()
---
include/aidge/backend/opencv/data/TensorImpl.hpp | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/include/aidge/backend/opencv/data/TensorImpl.hpp b/include/aidge/backend/opencv/data/TensorImpl.hpp
index 2765c94..708f196 100644
--- a/include/aidge/backend/opencv/data/TensorImpl.hpp
+++ b/include/aidge/backend/opencv/data/TensorImpl.hpp
@@ -94,6 +94,10 @@ public:
inline std::size_t scalarSize() const noexcept override final { return sizeof(T); }
+ void zeros() override final {
+ mData.setTo(Scalar::all(T(0)));
+ }
+
void copy(const void *src, NbElts_t length, NbElts_t offset = 0) override final {
const T* srcT = static_cast<const T *>(src);
T* dstT = static_cast<T *>(rawPtr(offset));
--
GitLab