15 lines
274 B
TypeScript
15 lines
274 B
TypeScript
import * as tf from '@tensorflow/tfjs-core';
|
|
|
|
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
|
|
)
|
|
)
|
|
} |