extractConvParamsFactory.js 627 B

12345678910
  1. import * as tf from '@tensorflow/tfjs-core';
  2. export function extractConvParamsFactory(extractWeights, paramMappings) {
  3. return function (channelsIn, channelsOut, filterSize, mappedPrefix) {
  4. var filters = tf.tensor4d(extractWeights(channelsIn * channelsOut * filterSize * filterSize), [filterSize, filterSize, channelsIn, channelsOut]);
  5. var bias = tf.tensor1d(extractWeights(channelsOut));
  6. paramMappings.push({ paramPath: mappedPrefix + "/filters" }, { paramPath: mappedPrefix + "/bias" });
  7. return { filters: filters, bias: bias };
  8. };
  9. }
  10. //# sourceMappingURL=extractConvParamsFactory.js.map