本文整理汇总了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);