本文整理匯總了TypeScript中deeplearn.NDArrayMath.matMul方法的典型用法代碼示例。如果您正苦於以下問題：TypeScript NDArrayMath.matMul方法的具體用法？TypeScript NDArrayMath.matMul怎麽用？TypeScript NDArrayMath.matMul使用的例子？那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類deeplearn.NDArrayMath的用法示例。
在下文中一共展示了NDArrayMath.matMul方法的2個代碼示例，這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚，您的評價將有助於係統推薦出更棒的TypeScript代碼示例。
示例1: run
async run(size: number): Promise<number> {
if (this.lastRunTimeMs > LAST_RUN_CPU_CUTOFF_MS) {
return new Promise<number>((resolve, reject) => {
resolve(-1);
});
}
const safeMode = false;
const math = new dl.NDArrayMath('cpu', safeMode);
dl.ENV.setMath(math);
const a: dl.Tensor2D = dl.randomUniform([size, size], -1, 1);
const b: dl.Tensor2D = dl.randomUniform([size, size], -1, 1);
const start = performance.now();
math.matMul(a, b);
const end = performance.now();
math.dispose();
this.lastRunTimeMs = end - start;
return this.lastRunTimeMs;
}示例2:
const benchmark = () => math.matMul(a, b);