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

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

.scalar()函数用于创建张量均值的标量类型。定标器是zero-dimension数组，也称为0级张量。使用.scalar()函数创建标量。

用法:

t.scalar( value, dataType )

参数：

value:标量的值。该值可以是数字，字符串，Uint8Array []，布尔值。
dataType [可选]：值的数据类型。它可以是int32，float32，bool，complex64或字符串。

返回值：它返回张量对象。

创建一个标量：在此示例中，我们将创建一个新的标量，表示仅一个值的张量。

范例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Value of a scalar
var value = 12
  
// Creating the value of a scalar
var tens = tf.scalar(value)
  
Printing the scalar
tens.print();

输出：

Tensor
    12

创建特定数据类型的标量：在此示例中，我们将创建特定数据类型的标量。请注意，数据类型只能是int32，float32，bool，complex64或string。

范例2：

Javascript


// Importing the tennsorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating a scalar using int value
var int_tensor = tf.scalar(12, 'int32')
  
int_tensor.print()
  
// Creating a scalar using string value
var str_tensor = tf.scalar("GFG", "string")
  
str_tensor.print()
  
// Creating a scalar using float value
var float_tensor = tf.scalar(12.6, "float32")
  
float_tensor.print();
  
// Creating a scalar using bool value
var bool_tensor1 = tf.scalar(true, "bool")
  
bool_tensor1.print()
  
// Creating a scalar using bool(0 and 1) type 
var bool_tensor2 = tf.scalar(0, "bool")
  
bool_tensor2.print()

输出：

Tensor
    12
Tensor
    GFG
Tensor
    12.600000381469727
Tensor
    true
Tensor
    false

注意：您也可以使用tf.tensor()函数创建标量。让我们看一个例子

使用tf.tensor()创建标量：

范例3：

Javascript


// Importing the tennsorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating a scalar using int tf.tensor()
var tens = tf.tensor(12, [], "int32")
  
tens.print()

在这里，我们为函数的第二个参数提供了一个空数组，因为我们正在创建一个标量，并且标量是rank-0张量。

输出：

Tensor
    12

相关用法

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