UNPKG

@hoff97/tensor-js

Version:

PyTorch like deep learning inferrence library

4 lines (3 loc) 289 B
import { CPUTensor } from '../../tensor/cpu/tensor'; import { DType } from '../../types'; export declare function gemm<DTpe extends DType>(a: CPUTensor<DTpe>, b: CPUTensor<DTpe>, aTranspose: boolean, bTranspose: boolean, alpha: number, beta: number, c?: CPUTensor<DTpe>): CPUTensor<DTpe>;