2020-08-20 02:05:34 +02:00
|
|
|
"use strict";
|
|
|
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
|
|
|
exports.extractFCParamsFactory = void 0;
|
|
|
|
|
const tf = require("@tensorflow/tfjs-core");
|
|
|
|
|
function extractFCParamsFactory(extractWeights, paramMappings) {
|
2020-08-18 14:04:33 +02:00
|
|
|
return function (channelsIn, channelsOut, mappedPrefix) {
|
|
|
|
|
const fc_weights = tf.tensor2d(extractWeights(channelsIn * channelsOut), [channelsIn, channelsOut]);
|
|
|
|
|
const fc_bias = tf.tensor1d(extractWeights(channelsOut));
|
|
|
|
|
paramMappings.push({ paramPath: `${mappedPrefix}/weights` }, { paramPath: `${mappedPrefix}/bias` });
|
|
|
|
|
return {
|
|
|
|
|
weights: fc_weights,
|
|
|
|
|
bias: fc_bias
|
|
|
|
|
};
|
|
|
|
|
};
|
|
|
|
|
}
|
2020-08-20 02:05:34 +02:00
|
|
|
exports.extractFCParamsFactory = extractFCParamsFactory;
|
2020-08-18 14:04:33 +02:00
|
|
|
//# sourceMappingURL=extractFCParamsFactory.js.map
|
