sharedLayers.js 762 B

123456789101112131415161718
  1. import * as tf from '@tensorflow/tfjs-core';
  2. import { convLayer } from '../common';
  3. import { prelu } from './prelu';
  4. export function sharedLayer(x, params, isPnet) {
  5. if (isPnet === void 0) { isPnet = false; }
  6. return tf.tidy(function () {
  7. var out = convLayer(x, params.conv1, 'valid');
  8. out = prelu(out, params.prelu1_alpha);
  9. out = tf.maxPool(out, isPnet ? [2, 2] : [3, 3], [2, 2], 'same');
  10. out = convLayer(out, params.conv2, 'valid');
  11. out = prelu(out, params.prelu2_alpha);
  12. out = isPnet ? out : tf.maxPool(out, [3, 3], [2, 2], 'valid');
  13. out = convLayer(out, params.conv3, 'valid');
  14. out = prelu(out, params.prelu3_alpha);
  15. return out;
  16. });
  17. }
  18. //# sourceMappingURL=sharedLayers.js.map