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

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

.tensor()函数用于在值，形状和数据类型的帮助下创建新的张量。

用法：

tf.tensor( value, shape, dataType)

Parameters:

Value:张量的值，它可以是数字的简单或嵌套Array或TypedArray。如果数组元素是字符串，则它们将编码为UTF-8并保持为Uint8Array []。
形状[可选]：它是一个可选参数。它采用张量的形状。如果未提供，则张量将从该值推断出其形状。
dataType [可选]：它也是可选参数。它可以是“ float32”或“ int32”或‘bool’或“ complex64”或‘string’。

返回值：它返回相同数据类型的张量。

范例1：在这个例子中，我们创建一个张量并打印它。为了创建张量，我们使用.tensor()方法，而要打印张量，我们使用.print()方法。

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
 
// Creating the tensor
var val = tf.tensor([1, 2, 3, 4, 5, 6, 7]);
 
// Printing the tensor
val.print()

输出：

Tensor
    [1, 2, 3, 4, 5, 6, 7]

范例2：在此示例中，我们创建的张量没有提到张量的shape参数，让我们在这里看到shape参数。

Javascript


// Importing the tensorflow library
import * as tf from "@tensorflow/tfjs"
 
// Defining the value of the tensor
var value = [1, 2, 3, 4, 5, 6]    
 
// Specify the shape of the tensor
var shape = [2, 3]
 
// Creting the tensor
var val = tf.tensor(value, shape)
 
// Printing the tensor
val.print()

输出：

Tensor
    [[1, 2, 3],
     [4, 5, 6]]

上面的示例创建了2×3维的张量。

范例3：在此示例中，我们将创建具有值，形状和数据类型的张量。我们正在创建String类型值的张量。

Javascript


// Importing the tensorflow.Js lbrary
import * as tf from "@tensorflow/tfjs"
 
// Creting a value variable which
// stores the value
var value = ['1', '2', '3', '4', '5', '6']
 
// Creting a shape variable
// which stores the shape
var shape = [2, 3]
 
// Creating a d_Type variable
// which strores the data-type
var d_Type = 'string'
 
// Creating the tensor
var val = tf.tensor(value, shape, d_Type)
 
// Printing the tensor
val.print()

输出：

Tensor
    [['1', '2', '3'],
     ['4', '5', '6']]

打印String类型值的张量。

相关用法

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