extractFCParamsFactory.js 605 B

12345678910111213
  1. import * as tf from '@tensorflow/tfjs-core';
  2. export function extractFCParamsFactory(extractWeights, paramMappings) {
  3. return function (channelsIn, channelsOut, mappedPrefix) {
  4. var fc_weights = tf.tensor2d(extractWeights(channelsIn * channelsOut), [channelsIn, channelsOut]);
  5. var fc_bias = tf.tensor1d(extractWeights(channelsOut));
  6. paramMappings.push({ paramPath: mappedPrefix + "/weights" }, { paramPath: mappedPrefix + "/bias" });
  7. return {
  8. weights: fc_weights,
  9. bias: fc_bias
  10. };
  11. };
  12. }
  13. //# sourceMappingURL=extractFCParamsFactory.js.map