tensorflow-helpers/tensor.d.ts

Helper functions to use tensorflow in nodejs for transfer learning, image classification, and more

import * as tf from '@tensorflow/tfjs-core';
export declare function disposeTensor(tensor: tf.Tensor | tf.Tensor[]): void;
/** @deprecated recommend to explore the model output shape at dev time, and use tf.squeeze() when needed */
export declare function toOneTensor(tensor: tf.Tensor | tf.Tensor[] | tf.NamedTensorMap): tf.Tensor;
export declare function toTensor4D(tensor: tf.Tensor3D | tf.Tensor4D): tf.Tensor4D;
export declare function toTensor3D(tensor: tf.Tensor3D | tf.Tensor4D): tf.Tensor3D;