diff --git a/src/lib/matrix/matrix/Matrix.hpp b/src/lib/matrix/matrix/Matrix.hpp index d17447b5b2..8df9ede9d9 100644 --- a/src/lib/matrix/matrix/Matrix.hpp +++ b/src/lib/matrix/matrix/Matrix.hpp @@ -604,6 +604,20 @@ public: } } + inline void nanToZero() + { + + Matrix &self = *this; + + for (size_t i = 0; i < M; i++) { + for (size_t j = 0; j < N; j++) { + if (std::isnan(self(i, j))) { + self(i, j) = static_cast(0.0f); + } + } + } + } + Matrix abs() const { Matrix r;