gx

chenyc

2025-06-12 7b72ac13a83764a662159d4a49b7fffb90476ecb

1
2
3
4
5
6
7
8
9
10
11
12
13
import * as tf from '../../dist/tfjs.esm';
 
import { ConvParams } from './types';
 
// eslint-disable-next-line no-unused-vars
export function loadConvParamsFactory(extractWeightEntry: <T>(originalPath: string, paramRank: number) => T) {
  return (prefix: string): ConvParams => {
    const filters = extractWeightEntry<tf.Tensor4D>(`${prefix}/filters`, 4);
    const bias = extractWeightEntry<tf.Tensor1D>(`${prefix}/bias`, 1);
 
    return { filters, bias };
  };
}