@tensorflow/tfjs-layers
Version:
TensorFlow layers API in JavaScript
30 lines (29 loc) • 1.23 kB
TypeScript
/**
* @license
* Copyright 2018 Google LLC
*
* Use of this source code is governed by an MIT-style
* license that can be found in the LICENSE file or at
* https://opensource.org/licenses/MIT.
* =============================================================================
*/
/// <amd-module name="@tensorflow/tfjs-layers/dist/utils/conv_utils" />
import { PaddingMode } from '../keras_format/common';
/**
* Transforms a single number of array of numbers into an array of numbers.
* @param value
* @param n: The size of the tuple to be returned.
* @param name: Name of the parameter, used for generating error messages.
* @returns An array of numbers.
*/
export declare function normalizeArray(value: number | number[], n: number, name: string): number[];
/**
* Determines output length of a convolution given input length.
* @param inputLength
* @param filterSize
* @param padding
* @param stride
* @param dilation: dilation rate.
*/
export declare function convOutputLength(inputLength: number, filterSize: number, padding: PaddingMode, stride: number, dilation?: number): number;
export declare function deconvLength(dimSize: number, strideSize: number, kernelSize: number, padding: PaddingMode): number;