convLayer.js 727 B

1234567891011121314151617181920
  1. import * as tf from '@tensorflow/tfjs-core';
  2. import { scale } from './scaleLayer';
  3. function convLayer(x, params, strides, withRelu, padding) {
  4. if (padding === void 0) { padding = 'same'; }
  5. var _a = params.conv, filters = _a.filters, bias = _a.bias;
  6. var out = tf.conv2d(x, filters, strides, padding);
  7. out = tf.add(out, bias);
  8. out = scale(out, params.scale);
  9. return withRelu ? tf.relu(out) : out;
  10. }
  11. export function conv(x, params) {
  12. return convLayer(x, params, [1, 1], true);
  13. }
  14. export function convNoRelu(x, params) {
  15. return convLayer(x, params, [1, 1], false);
  16. }
  17. export function convDown(x, params) {
  18. return convLayer(x, params, [2, 2], true, 'valid');
  19. }
  20. //# sourceMappingURL=convLayer.js.map