@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
11 lines (10 loc) • 481 B
TypeScript
import { Mode } from '../../../model/module';
import Tensor, { DType } from '../../../types';
import { Attributes, Constants } from '../../types';
import { UnaryNode } from './unaryNode';
export declare class HardSigmoidNode extends UnaryNode {
alpha: number;
beta: number;
constructor(attributes: Attributes, inputs: string[], outputs: string[], constants: Constants, onnxVersion: number, mode: Mode);
compute<DTpe extends DType>(x: Tensor<DTpe>): Tensor<DTpe>;
}