added ones function

m-a-d-n-e-s-s · Oct 16, 2023 · b7e8639 · b7e8639
1 parent 5f96b4d
commit b7e8639
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/madness/tensor/tensor.h b/src/madness/tensor/tensor.h
@@ -1955,6 +1955,14 @@ MADNESS_PRAGMA_GCC(diagnostic pop)
         return (-t)+=x;
     }
 
+    /// Returns a new tensor filled with ones
+    template <typename T>
+    Tensor<T> ones(long n) {
+        Tensor<T> result = Tensor<T>(1,&n,false);
+        UNARY_OPTIMIZED_ITERATOR(T,result,*_p0 = 1.0);
+        return result;
+    }
+
     /// Returns a new contiguous tensor that is a deep copy of the input
 
     /// \ingroup tensor