2020-10-09 03:31:31 +02:00
|
|
|
import * as tf from '@tensorflow/tfjs';
|
2020-08-18 13:54:53 +02:00
|
|
|
|
|
|
|
|
import { ScaleLayerParams } from './types';
|
|
|
|
|
|
|
|
|
|
export function scale(x: tf.Tensor4D, params: ScaleLayerParams): tf.Tensor4D {
|
|
|
|
|
return tf.add(tf.mul(x, params.weights), params.biases)
|
|
|
|
|
}
|
