UNPKG

1.46 kBTypeScriptView Raw
1/**
2 * @license
3 * Copyright 2020 Google LLC. All Rights Reserved.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 * =============================================================================
16 */
17/// <amd-module name="@tensorflow/tfjs-core/dist/ops/rand" />
18import { Tensor } from '../tensor';
19import { DataType, Rank, ShapeMap } from '../types';
20/**
21 * Creates a `tf.Tensor` with values sampled from a random number generator
22 * function defined by the user.
23 *
24 * @param shape An array of integers defining the output tensor shape.
25 * @param randFunction A random number generator function which is called
26 * for each element in the output tensor.
27 * @param dtype The data type of the output tensor. Defaults to 'float32'.
28 *
29 * @doc {heading: 'Tensors', subheading: 'Random'}
30 */
31declare function rand_<R extends Rank>(shape: ShapeMap[R], randFunction: () => number, dtype?: DataType): Tensor<R>;
32export declare const rand: typeof rand_;
33export {};