package com.sparktest.demo
import breeze.linalg.{Axis, DenseMatrix, DenseVector, accumulate, sum, trace}
/**
* Created by HeartisTiger on 2019/7/8
*/
object Demo05 {
def main(args: Array[String]): Unit = {
var m1 = new DenseMatrix[Int](3,3,Array(1,2,3,4,5,6,7,8,9))
val v1 = DenseVector(1,2,3,4,5,6)
//求和
println(sum(m1))
//每一列求和
println(sum(m1, Axis._0))
//每一行求和
println(sum(m1, Axis._1))
//对角线元素和
println(trace(m1))
//向量累积和
println(accumulate(v1))
}
}
Spark机器学习库-矩阵or向量的求和函数
最新推荐文章于 2022-04-10 15:33:37 发布