UNPKG

@tensorflow/tfjs-core

Version:

Hardware-accelerated JavaScript library for machine intelligence

97 lines (96 loc) 3.08 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 } from '../tensor'; /** * Fast Fourier transform. * * Computes the 1-dimensional discrete Fourier transform over the inner-most * dimension of input. * * ```js * const real = tf.tensor1d([1, 2, 3]); * const imag = tf.tensor1d([1, 2, 3]); * const x = tf.complex(real, imag); * * x.fft().print(); // tf.spectral.fft(x).print(); * ``` * @param input The complex input to compute an fft over. */ /** * @doc {heading: 'Operations', subheading: 'Spectral', namespace: 'spectral'} */ declare function fft_(input: Tensor): Tensor; /** * Inverse fast Fourier transform. * * Computes the inverse 1-dimensional discrete Fourier transform over the * inner-most dimension of input. * * ```js * const real = tf.tensor1d([1, 2, 3]); * const imag = tf.tensor1d([1, 2, 3]); * const x = tf.complex(real, imag); * * x.ifft().print(); // tf.spectral.ifft(x).print(); * ``` * @param input The complex input to compute an ifft over. */ /** * @doc {heading: 'Operations', subheading: 'Spectral', namespace: 'spectral'} */ declare function ifft_(input: Tensor): Tensor; /** * Real value input fast Fourier transform. * * Computes the 1-dimensional discrete Fourier transform over the * inner-most dimension of the real input. * * ```js * const real = tf.tensor1d([1, 2, 3]); * * real.rfft().print(); * ``` * @param input The real value input to compute an rfft over. */ /** * @doc {heading: 'Operations', subheading: 'Spectral', namespace: 'spectral'} */ declare function rfft_(input: Tensor, fftLength?: number): Tensor; /** * Inversed real value input fast Fourier transform. * * Computes the 1-dimensional inversed discrete Fourier transform over the * inner-most dimension of the real input. * * ```js * const real = tf.tensor1d([1, 2, 3]); * const imag = tf.tensor1d([0, 0, 0]); * const x = tf.complex(real, imag); * * x.irfft().print(); * ``` * @param input The real value input to compute an irfft over. */ /** * @doc {heading: 'Operations', subheading: 'Spectral', namespace: 'spectral'} */ declare function irfft_(input: Tensor): Tensor; export declare const fft: typeof fft_; export declare const ifft: typeof ifft_; export declare const rfft: typeof rfft_; export declare const irfft: typeof irfft_; export {};