diff --git a/mllib/src/main/scala/org/apache/spark/mllib/linalg/Vectors.scala b/mllib/src/main/scala/org/apache/spark/mllib/linalg/Vectors.scala index 91f065831c80..cc8ad8792b21 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/linalg/Vectors.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/linalg/Vectors.scala @@ -728,6 +728,20 @@ object DenseVector { def fromML(v: newlinalg.DenseVector): DenseVector = { new DenseVector(v.values) } + + + /** + * Generate a `DenseVector` consisting of zeros. + * + * @param size size of the DenseVector + * @return a `DenseVector` consisting of zeros and Double type. + */ + @Since("2.0.0") + def zeros(size: Int): DenseVector = { + val bdv=BDV.zeros[Double](size) + return new DenseVector(bdv.toArray) + } + } /**