Tensorflow.js tf.Tensor.print()用法及代码示例

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

tf.print() 函数用于打印有关 tf.Tensor 的信息，包括其数据。

用法:

tf.print(value, verbose)

参数：

• value:张量的值可以是一个简单的或嵌套的 Array 或 TypedArray 的数字。如果数组元素是字符串，则它们将编码为 UTF-8 并保留为 Uint8Array[]。
• verbose:它是一个布尔值，用于指示是否打印有关 Tensor 的详细信息，包括 dtype 和大小，verbose 的默认值为 False。

以下示例演示了 tf.print() 函数：

范例1：在这个例子中，我们使用 ts.tensor2d 创建一个张量，我们使用 tf.print 函数使用详细值作为 true 进行打印。

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating the tensor
const verbose = true;
var val = tf.tensor2d(["geeks","for","geeks","website"], [2, 2]);
  
// Printing the tensor
val.print(verbose);

输出：

Tensor
  dtype:string
  rank:2
  shape:[2,2]
  values:
    [['geeks', 'for'    ],
     ['geeks', 'website']]

范例2：在这个例子中，我们使用 ts.tensor2d 创建一个张量，我们使用 tf.print 函数使用详细值作为 false 进行打印。

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating the tensor
const verbose = false;
var val = tf.tensor(["geeks","for","geeks"]);
  
// Printing the tensor
val.print(verbose);

输出：

Tensor
    ['geeks', 'for', 'geeks']

参考：https://js.tensorflow.org/api/latest/#tf.Tensor.print