convLayer.js 883 B

12345678910111213141516171819202122232425
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", { value: true });
  3. var tf = require("@tensorflow/tfjs-core");
  4. var scaleLayer_1 = require("./scaleLayer");
  5. function convLayer(x, params, strides, withRelu, padding) {
  6. if (padding === void 0) { padding = 'same'; }
  7. var _a = params.conv, filters = _a.filters, bias = _a.bias;
  8. var out = tf.conv2d(x, filters, strides, padding);
  9. out = tf.add(out, bias);
  10. out = scaleLayer_1.scale(out, params.scale);
  11. return withRelu ? tf.relu(out) : out;
  12. }
  13. function conv(x, params) {
  14. return convLayer(x, params, [1, 1], true);
  15. }
  16. exports.conv = conv;
  17. function convNoRelu(x, params) {
  18. return convLayer(x, params, [1, 1], false);
  19. }
  20. exports.convNoRelu = convNoRelu;
  21. function convDown(x, params) {
  22. return convLayer(x, params, [2, 2], true, 'valid');
  23. }
  24. exports.convDown = convDown;
  25. //# sourceMappingURL=convLayer.js.map