19 lines
665 B
JavaScript
19 lines
665 B
JavaScript
|
import * as tf from '@tensorflow/tfjs-core';
|
||
|
|
import { scale } from './scaleLayer';
|
||
|
|
function convLayer(x, params, strides, withRelu, padding = 'same') {
|
||
|
|
const { filters, bias } = params.conv;
|
||
|
|
let out = tf.conv2d(x, filters, strides, padding);
|
||
|
|
out = tf.add(out, bias);
|
||
|
|
out = scale(out, params.scale);
|
||
|
|
return withRelu ? tf.relu(out) : out;
|
||
|
|
}
|
||
|
|
export function conv(x, params) {
|
||
|
|
return convLayer(x, params, [1, 1], true);
|
||
|
|
}
|
||
|
|
export function convNoRelu(x, params) {
|
||
|
|
return convLayer(x, params, [1, 1], false);
|
||
|
|
}
|
||
|
|
export function convDown(x, params) {
|
||
|
|
return convLayer(x, params, [2, 2], true, 'valid');
|
||
|
|
}
|
||
|
|
//# sourceMappingURL=convLayer.js.map
|