UNPKG

@tensorflow-models/coco-ssd

Version:

Object detection model (coco-ssd) in TensorFlow.js

86 lines (85 loc) 3.1 kB
/** * @license * Copyright 2018 Google Inc. All Rights Reserved. * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. * ============================================================================= */ import { Tensor, Tensor1D, Tensor2D, Tensor3D, Tensor4D } from '../tensor'; import { TensorLike } from '../types'; /** * Reverses a `tf.Tensor1D`. * * @param x The input tensor. */ declare function reverse1d_(x: Tensor1D | TensorLike): Tensor1D; /** * Reverses a `tf.Tensor2D` along a specified axis. * * @param x The input tensor. * @param axis The set of dimensions to reverse. Must be in the * range [-rank(x), rank(x)). Defaults to all axes. */ declare function reverse2d_(x: Tensor2D | TensorLike, axis?: number | number[]): Tensor2D; /** * Reverses a `tf.Tensor3D` along a specified axis. * * @param x The input tensor. * @param axis The set of dimensions to reverse. Must be in the * range [-rank(x), rank(x)). Defaults to all axes. */ declare function reverse3d_(x: Tensor3D | TensorLike, axis?: number | number[]): Tensor3D; /** * Reverses a `tf.Tensor4D` along a specified axis. * * @param x The input tensor. * @param axis The set of dimensions to reverse. Must be in the * range [-rank(x), rank(x)). Defaults to all axes. */ declare function reverse4d_(x: Tensor4D | TensorLike, axis?: number | number[]): Tensor4D; /** * Reverses a `tf.Tensor` along a specified axis. * * Also available are stricter rank-specific methods that assert that `x` is * of the given rank: * - `tf.reverse1d` * - `tf.reverse2d` * - `tf.reverse3d` * - `tf.reverse4d` * * Except `tf.reverse1d` (which does not have axis param), all methods have * same signature as this method. * * ```js * const x = tf.tensor1d([1, 2, 3, 4]); * * x.reverse().print(); * ``` * * ```js * const x = tf.tensor2d([1, 2, 3, 4], [2, 2]); * * const axis = 1; * x.reverse(axis).print(); * ``` * @param x The input tensor to be reversed. * @param axis The set of dimensions to reverse. Must be in the * range [-rank(x), rank(x)). Defaults to all axes. */ /** @doc {heading: 'Tensors', subheading: 'Slicing and Joining'} */ declare function reverse_<T extends Tensor>(x: T | TensorLike, axis?: number | number[]): T; export declare const reverse: typeof reverse_; export declare const reverse1d: typeof reverse1d_; export declare const reverse2d: typeof reverse2d_; export declare const reverse3d: typeof reverse3d_; export declare const reverse4d: typeof reverse4d_; export {};