1
2
3
4
5
6
7
8
9
10
11
12
13
| 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,
| ));
| }
|
|
