dotProductGPU function

GPUTensor<Scalar> dotProductGPU(
  1. GPUTensor<Vector> a,
  2. GPUTensor<Vector> b,
  3. CommandBuffer tape
)

Implementation

GPUTensor<Scalar> dotProductGPU(GPUTensor<Vector> a, GPUTensor<Vector> b, CommandBuffer tape) {
  GPUTensor<Vector> multiplied = multiplyGPU<Vector>(a, b, tape);
  GPUTensor<Scalar> out = sumGPU(multiplied, tape);
  return out;
}