UNPKG

@tensorflow/tfjs-core

Version:

Hardware-accelerated JavaScript library for machine intelligence

56 lines (55 loc) 2.02 kB
import { Tensor } from '../tensor'; import { TensorLike } from '../types'; /** * Converts two real numbers to a complex number. * * Given a tensor `real` representing the real part of a complex number, and a * tensor `imag` representing the imaginary part of a complex number, this * operation returns complex numbers elementwise of the form [r0, i0, r1, i1], * where r represents the real part and i represents the imag part. * * The input tensors real and imag must have the same shape. * * ```js * const real = tf.tensor1d([2.25, 3.25]); * const imag = tf.tensor1d([4.75, 5.75]); * const complex = tf.complex(real, imag); * * complex.print(); * ``` */ /** @doc {heading: 'Tensors', subheading: 'Creation'} */ declare function complex_<T extends Tensor>(real: T | TensorLike, imag: T | TensorLike): T; /** * Returns the real part of a complex (or real) tensor. * * Given a tensor input, this operation returns a tensor of type float that is * the real part of each element in input considered as a complex number. * * If the input is real, it simply makes a clone. * * ```js * const x = tf.complex([-2.25, 3.25], [4.75, 5.75]); * tf.real(x).print(); * ``` */ /** @doc {heading: 'Tensors', subheading: 'Creation'} */ declare function real_<T extends Tensor>(input: T | TensorLike): T; /** * Returns the imaginary part of a complex (or real) tensor. * * Given a tensor input, this operation returns a tensor of type float that is * the imaginary part of each element in input considered as a complex number. * If input is real, a tensor of all zeros is returned. * * ```js * const x = tf.complex([-2.25, 3.25], [4.75, 5.75]); * tf.imag(x).print(); * ``` */ /** @doc {heading: 'Tensors', subheading: 'Creation'} */ declare function imag_<T extends Tensor>(input: T | TensorLike): T; export declare const complex: typeof complex_; export declare const real: typeof real_; export declare const imag: typeof imag_; export {};