1
2
3
4
5
6
7
8
9
10
11
12
13
| 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
|
|
