@shumai/shumai/dist/shumai/module/norm.d.ts

A fast, network-connected, differentiable tensor library for TypeScript (and JavaScript). Built with bun + flashlight for software engineers and researchers alike.

import type { Tensor } from '../tensor';
import { Module } from './module';
export declare class LayerNorm extends Module {
    private dims;
    private axes;
    private eps;
    private gamma;
    private beta;
    constructor(dims: number[], eps?: number | Tensor);
    resetParameters(): void;
    forward(tensor: Tensor): Tensor;
}