1 | 'use strict'
|
2 |
|
3 | const size = require('../../utils/array').size
|
4 |
|
5 | function factory (type, config, load, typed) {
|
6 | const add = load(require('../arithmetic/add'))
|
7 | const multiply = load(require('../arithmetic/multiply'))
|
8 |
|
9 | |
10 |
|
11 |
|
12 |
|
13 |
|
14 |
|
15 |
|
16 |
|
17 |
|
18 |
|
19 |
|
20 |
|
21 |
|
22 |
|
23 |
|
24 |
|
25 |
|
26 |
|
27 |
|
28 |
|
29 |
|
30 |
|
31 |
|
32 | const dot = typed('dot', {
|
33 | 'Matrix, Matrix': function (x, y) {
|
34 | return _dot(x.toArray(), y.toArray())
|
35 | },
|
36 |
|
37 | 'Matrix, Array': function (x, y) {
|
38 | return _dot(x.toArray(), y)
|
39 | },
|
40 |
|
41 | 'Array, Matrix': function (x, y) {
|
42 | return _dot(x, y.toArray())
|
43 | },
|
44 |
|
45 | 'Array, Array': _dot
|
46 | })
|
47 |
|
48 | dot.toTex = { 2: `\\left(\${args[0]}\\cdot\${args[1]}\\right)` }
|
49 |
|
50 | return dot
|
51 |
|
52 | |
53 |
|
54 |
|
55 |
|
56 |
|
57 |
|
58 |
|
59 |
|
60 | function _dot (x, y) {
|
61 | const xSize = size(x)
|
62 | const ySize = size(y)
|
63 | const len = xSize[0]
|
64 |
|
65 | if (xSize.length !== 1 || ySize.length !== 1) throw new RangeError('Vector expected')
|
66 | if (xSize[0] !== ySize[0]) throw new RangeError('Vectors must have equal length (' + xSize[0] + ' != ' + ySize[0] + ')')
|
67 | if (len === 0) throw new RangeError('Cannot calculate the dot product of empty vectors')
|
68 |
|
69 | let prod = 0
|
70 | for (let i = 0; i < len; i++) {
|
71 | prod = add(prod, multiply(x[i], y[i]))
|
72 | }
|
73 |
|
74 | return prod
|
75 | }
|
76 | }
|
77 |
|
78 | exports.name = 'dot'
|
79 | exports.factory = factory
|