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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。