Tag: gpu

Here is one approach-

def normalize(vectors: List[(String, INDArray)]) = {
    vectors.map(
      (value: (String, INDArray)) => {
        val vec: INDArray = value._2
        val vec2: INDArray = vec.mul(vec)
        val total = vec2.sumNumber()

        value._2.div(sqrt(total.doubleValue()))
      }
    )
  }