Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Scala Breeze Cartesian product of vectors

Let's say I have 2 vectors of the same dimensionality:

scala> val v = DenseVector(3.0, 4.0)
v: breeze.linalg.DenseVector[Double] = DenseVector(3.0, 4.0)

scala> val w = DenseVector(5.0, 6.0)
w: breeze.linalg.DenseVector[Double] = DenseVector(5.0, 6.0)

How can I compute a matrix from it, containing all products (something like a cartesian product in db speak):

scala> val x = DenseMatrix(
     | (v(0) * w(0), v(0) * w(1)),
     | (v(1) * w(0), v(1) * w(1))
     | )
x: breeze.linalg.DenseMatrix[Double] =
15.0  18.0
20.0  24.0

I sense you should be able to do it with broadcasting, but I can't figure out how. Thank you.

like image 497
botkop Avatar asked Sep 18 '25 15:09

botkop


1 Answers

This is just the outer product of two vectors:

scala> import breeze.linalg._
import breeze.linalg._

scala> val v = DenseVector(3.0, 4.0)
v: breeze.linalg.DenseVector[Double] = DenseVector(3.0, 4.0)

scala>  val w = DenseVector(5.0, 6.0)
w: breeze.linalg.DenseVector[Double] = DenseVector(5.0, 6.0)

scala> v * w.t
res0: breeze.linalg.DenseMatrix[Double] =
15.0  18.0
20.0  24.0
like image 117
dlwh Avatar answered Sep 21 '25 09:09

dlwh