fullyConnectedLayer.js 250 B

1234567
  1. import * as tf from '@tensorflow/tfjs-core';
  2. export function fullyConnectedLayer(x, params) {
  3. return tf.tidy(function () {
  4. return tf.add(tf.matMul(x, params.weights), params.bias);
  5. });
  6. }
  7. //# sourceMappingURL=fullyConnectedLayer.js.map