Tensorflow.js tf.util.flatten()函数

Tensorflow.js 是一个由谷歌开发的开源库，用于在浏览器或节点环境中运行机器学习模型以及深度学习神经网络。

.util.flatten()函数用于展平嵌套的不一致数组。

句法 ：

tf.util.flatten(arr, result?, skipTypedArray?)

参数：

• arr：要展平的嵌套数组。它可以是 number、Boolean、 字符串、Promise、TypedArray、RecursiveArray 或 TypedArray> 类型。
• 结果：它是携带元素的指定目标数组。它是可选参数，可以是 number、Boolean、 字符串、Promise、TypedArray[] 类型。
• skipTypedArray：它是防止类型化数组展平的可选参数。它的默认值为false 。

返回值：可以返回 number、Boolean、 字符串、Promise 或 TypedArray[]。

示例 1：

Javascript

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining nested array
const arr = [[11, 12], [13, 14], [15, [16, [17]]]];
  
// Calling tf.util.flatten() method
const res = tf.util.flatten(arr);
  
// printing output
console.log(res);

输出：

11,12,13,14,15,16,17

示例 2：

Javascript

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining nested array
const arr = [[11, 12], [13, 14], [15, [16, [17]]]];
  
// Defining destination array
const des_arr = [9, 10]
  
// Calling tf.util.flatten() method with
// all its parameters
const res = tf.util.flatten(arr, des_arr, true);
  
// printing output
console.log(res);

输出：

9,10,11,12,13,14,15,16,17

参考： https://js.tensorflow.org/api/1.0.0/#flatten