13 lines
No EOL
617 B
JavaScript
13 lines
No EOL
617 B
JavaScript
import * as tf from '@tensorflow/tfjs-core';
|
|
export function extractFCParamsFactory(extractWeights, paramMappings) {
|
|
return function (channelsIn, channelsOut, mappedPrefix) {
|
|
var fc_weights = tf.tensor2d(extractWeights(channelsIn * channelsOut), [channelsIn, channelsOut]);
|
|
var fc_bias = tf.tensor1d(extractWeights(channelsOut));
|
|
paramMappings.push({ paramPath: mappedPrefix + "/weights" }, { paramPath: mappedPrefix + "/bias" });
|
|
return {
|
|
weights: fc_weights,
|
|
bias: fc_bias
|
|
};
|
|
};
|
|
}
|
|
//# sourceMappingURL=extractFCParamsFactory.js.map
|