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

Tensorflow.js是Google开发的开源库，用于在浏览器或节点环境中运行机器学习模型和深度学习神经网络。它可以帮助开发人员使用JavaScript开发ML模型，并直接在浏览器或Node.js中使用ML。

tf.print()函数用于打印有关tf.tensor的信息及其数据。

用法:

tf.print (x, verbose)

参数：

x:要打印的张量
verbose:它是一个可选参数。这些是布尔参数，它决定是否打印有关Tensor的详细信息，包括dtype和size。

返回：它不返回任何内容，即无效。

范例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Setting the value of verbose
const verbose = true;
  
// Creating the tensor
var val = tf.tensor2d([8, 2, 5, 6], [2, 2]);
  
// Printing the tensor
val.print(verbose)

输出：

Tensor
   dtype:float32
   rank:2
   shape:[2,2]
   values:
      [[8, 2],
       [5, 6]]

范例2：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Setting the value of verbose
const verbose = false;
  
// Creating the tensor
var val = tf.tensor2d([8, 2, 5, 6], [2, 2]);
  
// Printing the tensor
val.print(verbose)

输出：

Tensor
   [[8, 2],
    [5, 6]]

参考：https://js.tensorflow.org/api/3.4.0/#print

相关用法

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