@tensorflow/tfjs-core
Version:
Hardware-accelerated JavaScript library for machine intelligence
8 lines (7 loc) • 463 B
TypeScript
/// <amd-module name="@tensorflow/tfjs-core/dist/public/chained_ops/batchnorm" />
import { Rank, TensorLike } from '../../types';
declare module '../../tensor' {
interface Tensor<R extends Rank = Rank> {
batchNorm<T extends Tensor>(mean: Tensor<R> | Tensor1D | TensorLike, variance: Tensor<R> | Tensor1D | TensorLike, offset?: Tensor<R> | Tensor1D | TensorLike, scale?: Tensor<R> | Tensor1D | TensorLike, varianceEpsilon?: number): Tensor<R>;
}
}