diff --git a/nntrainer/tensor/tensor_v2.cpp b/nntrainer/tensor/tensor_v2.cpp index 6e0f99d465..55c25af0a9 100644 --- a/nntrainer/tensor/tensor_v2.cpp +++ b/nntrainer/tensor/tensor_v2.cpp @@ -263,7 +263,7 @@ const std::array TensorV2::getStrides() const } bool TensorV2::checkContinuous(unsigned int np1, unsigned int np2) const { - if (np1 < 0 || np1 > 3 || np2 < 0 || np2 > 3) { + if (np1 > 3 || np2 > 3) { throw std::invalid_argument( "Error: Input value must be within the range of 0 to 3."); }