Dot product

[nicolaspanel]

Hi,

Is there a way/module to perform dot product of 2 ndarrays?

Best regards

[nicolaspanel]

ndgemm package handles matrix multiplication

[fasiha]

Consider adding .dot() as a user-friendly wrapper over ndgemm?

[fasiha]

This is about what I could manage:

function dot(...args) {
  return args.reduce((sofar, curr) => {
    const result = zeros([ sofar.shape[0], curr.shape[1] ]);
    gemm(result, sofar, curr);
    return result;
  });
}

Test code:

var zeros = require('zeros');
var ops = require('ndarray-ops');
var gemm = require('ndarray-gemm');

var a = ops.random(zeros([ 2, 3 ]));
var b = ops.random(zeros([ 3, 4 ]));
var c = ops.random(zeros([ 4, 2 ]));

var tmp = zeros([ 2, 4 ]);
var res1 = zeros([ 2, 2 ]);
gemm(tmp, a, b);
gemm(res1, tmp, c);

var res2 = dot(a, b, c);

assert(ops.equals(res1, res2));