1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
| /// <amd-module name="@tensorflow/tfjs-core/dist/ops/concat_3d" />
| /**
| * @license
| * Copyright 2020 Google LLC. 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 { Tensor3D } from '../tensor';
| import { TensorLike } from '../types';
| /**
| * Concatenates a list of `tf.Tensor3D`s along an axis.
| * See `concat` for details.
| *
| * For example, if:
| * A: shape(2, 1, 3) = | r1, g1, b1 |
| * | r2, g2, b2 |
| *
| * B: shape(2, 1, 3) = | r3, g3, b3 |
| * | r4, g4, b4 |
| *
| * C = tf.concat3d([A, B], axis)
| *
| * if axis = 0:
| * C: shape(4, 1, 3) = | r1, g1, b1 |
| * | r2, g2, b2 |
| * | r3, g3, b3 |
| * | r4, g4, b4 |
| *
| * if axis = 1:
| * C: shape(2, 2, 3) = | r1, g1, b1, r3, g3, b3 |
| * | r2, g2, b2, r4, g4, b4 |
| *
| * if axis = 2:
| * C = shape(2, 1, 6) = | r1, g1, b1, r3, g3, b3 |
| * | r2, g2, b2, r4, g4, b4 |
| *
| * @param tensors A list of`tf.Tensor`s to concatenate.
| * @param axis The axis to concate along.
| * @return The concatenated array.
| */
| declare function concat3d_(tensors: Array<Tensor3D | TensorLike>, axis: number): Tensor3D;
| export declare const concat3d: typeof concat3d_;
| export {};
|
|
