@vladmandic/face-api/src/common/fullyConnectedLayer.ts

FaceAPI: AI-powered Face Detection & Rotation Tracking, Face Description & Recognition, Age & Gender & Emotion Prediction for Browser and NodeJS using TensorFlow/JS

import * as tf from '../../dist/tfjs.esm';

import { FCParams } from './types';

export function fullyConnectedLayer(
  x: tf.Tensor2D,
  params: FCParams,
): tf.Tensor2D {
  return tf.tidy(() => tf.add(
    tf.matMul(x, params.weights),
    params.bias,
  ));
}