10 lines
616 B
JavaScript
10 lines
616 B
JavaScript
import * as tf from '@tensorflow/tfjs-core';
|
|
export function extractConvParamsFactory(extractWeights, paramMappings) {
|
|
return function (channelsIn, channelsOut, filterSize, mappedPrefix) {
|
|
const filters = tf.tensor4d(extractWeights(channelsIn * channelsOut * filterSize * filterSize), [filterSize, filterSize, channelsIn, channelsOut]);
|
|
const bias = tf.tensor1d(extractWeights(channelsOut));
|
|
paramMappings.push({ paramPath: `${mappedPrefix}/filters` }, { paramPath: `${mappedPrefix}/bias` });
|
|
return { filters, bias };
|
|
};
|
|
}
|
|
//# sourceMappingURL=extractConvParamsFactory.js.map
|