我实现了以下代码来使用 for 循环计算加权平均值,我怎样才能更 func 编程风格并使用 map
和 zip
?
val aggAvg = (emb: Seq[Seq[Float]], weights: Seq[Float]) => {
val embSize = emb.head.size
val len = emb.size
(0 until embSize)
.map { i =>
(0 until len).map { j =>
emb(j)(i) * weights(j)
}.sum / weights.sum
}
}
例子: 给定
val emb: Seq[Seq[Float]] = Seq(Seq(1,2,3), Seq(4,5,6))
val weights: Seq[Float] = Seq(2, 8)
输出将是 Seq(3.4, 4.4, 5.4)
因为
(1 * 2 + 4 * 8)/(2 + 8) = 3.4
等等。
最佳答案
这是一种方式,虽然我不确定它是否是最优雅的
val aggAvg = (emb: Seq[Seq[Float]], weights: Seq[Float]) =>
emb.transpose.map((weights, _).zipped.map(_ * _).sum).map(_ / weights.sum)
res0: Seq[Float] = List(3.4, 4.4, 5.4)
关于scala - 在 2 个 for 循环中使用 map 和 zip 使其更具功能性,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/69623371/