Tensorflow.js tf.util.flatten()用法及代码示例

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

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

用法：

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

Parameters:

arr:声明的嵌套数组将被展平。它的类型可以是数字，布尔值，字符串，Promise <数字>，TypedArray，RecursiveArray或TypedArray>。
result:声明的目标数组承载元素。它是可选参数，可以是数字，布尔值，字符串，Promise <数字>和TypedArray []类型。
skipTypedArray:它是可选参数，可防止类型化数组变平。并且它的默认值是false。

返回值：它可以返回数字，布尔值，字符串，Promise <number>或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

相关用法

注：本文由纯净天空筛选整理自nidhi1352singh大神的英文原创作品 Tensorflow.js tf.util.flatten() Function。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。