@tensorflow/tfjs-core

Version:

Hardware-accelerated JavaScript library for machine intelligence

6 lines (5 loc) • 224 B

TypeScript

import { Tensor } from '../tensor'; import { TensorLike } from '../types'; declare function softmax_<T extends Tensor>(logits: T | TensorLike, dim?: number): T; export declare const softmax: typeof softmax_; export {};