l2NormGPU function

GPUTensor<Scalar> l2NormGPU(
  1. GPUTensor<Vector> v,
  2. CommandBuffer tape
)

Implementation

GPUTensor<Scalar> l2NormGPU(GPUTensor<Vector> v, CommandBuffer tape) {
  // sqrt( sum( v^2 ) )
  GPUTensor<Vector> squared = powGPU<Vector>(v, 2.0, tape);
  GPUTensor<Scalar> sumOfSquares = sumGPU(squared, tape);

  return sqrtGPU<Scalar>(sumOfSquares, tape);
}