Tensorflow.js tf.tensor()函数

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

.tensor()函数用于在value 、 shape和data type的帮助下创建一个新的张量。

句法 ：

tf.tensor( value, shape, dataType)

参数：

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

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

示例 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：在这个例子中，我们在创建张量的地方没有提到张量的形状参数，我们看这里的形状参数。

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]
 
// Creating the tensor
var val = tf.tensor(value, shape)
 
// Printing the tensor
val.print()

输出：

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

上面的例子是创建 2 × 3 维度的张量。

示例 3：在此示例中，我们正在创建一个具有value、 shape和dataType 的张量。我们正在创建字符串类型值的张量。

Javascript

// Importing the tensorflow.Js library
import * as tf from "@tensorflow/tfjs"
 
// Creating a value variable which
// stores the value
var value = ['1', '2', '3', '4', '5', '6']
 
// Creating a shape variable
// which stores the shape
var shape = [2, 3]
 
// Creating a d_Type variable
// which stores 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']]

打印字符串类型值的张量。