如何将函数应用于矩阵?

Question

这是我上一个问题的后续

现在我知道cats为Vector提供了一个Apply实例。所以我可以写：

import cats.implicits._
scala> val f: Int => Int = _ + 2
f: Int => Int = <function1>
scala> Vector(f) ap Vector(1, 2, 3)
res18: scala.collection.immutable.Vector[Int] = Vector(3, 4, 5)

现在我想知道如何f应用于定义为Vector[Vector[Int]的矩阵的所有元素。cats是否为矩阵提供Apply实例？

Answer 1

为了在Vector[Vector[Int]]上使用ap，你需要f提升到Vector[Vector[A => B]]。一种可能的方法是：

import cats.Apply
import cats.implicits._
val f: Int => Int = _ + 2
val vectorOfVectors = Apply[Vector] compose Apply[Vector]
val vec = Vector(Vector(1,2), Vector(3,4))
val res: Vector[Vector[Int]] = vectorOfVectors.ap(Vector(Vector(f)))(vec)

收益 率：

Vector(3, 4)
Vector(5, 6)

IMO使用Nested更好方法是：

在build.sbt中使用kind-projector：

addCompilerPlugin("org.spire-math" %% "kind-projector" % "0.9.4")

然后：

import cats.Functor
import cats.data.Nested
import cats.implicits._
val f: Int => Int = _ + 2
val vec = Vector(Vector(1,2), Vector(3,4))
val nested: Nested[Vector, Vector, Int] = Nested(vec)
val res: Nested[Vector, Vector, Int] = Functor[Nested[Vector, Vector, ?]].map(nested)(f)
val result: Vector[Vector[Int]] = res.value

请注意，使用Nested意味着我们可以完全不提升地应用f。