Tensorflow.js tf.tensor6d()函数

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

.tensor6d()函数用于创建一个新的6 维张量，其参数为值、形状和数据类型。

句法：

tf.tensor6d(value, shape?, dataType?)

参数：

• value ：张量的值，可以是嵌套的数字数组、平面数组或TypedArray 。
• shape ：它采用张量的形状。如果未提供，张量将从值推断其形状。它是一个可选参数。
• 数据类型：它可以是“ float32 ”或“ int32 ”或“ bool ”或“ complex64 ”或“ 字符串 ”值。它是一个可选参数。

返回值：返回相同数据类型的张量。返回的张量将始终是 6 维的。

注意：6d 张量功能也可以使用 tf.tensor()函数来实现，但使用 tf.tensor6d() 会使代码易于理解和阅读。

示例 1：在这里，我们正在创建一个 6d 张量并打印它。为了创建 6d 张量，我们使用 . tensor6d ()函数，我们使用 .print()函数来打印张量。在这里，我们将 6d 数组（即嵌套数组）传递给 value 参数。

Javascript

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs";
  
// Create the tensor
var example1 = tf.tensor6d([[[[
    [[1, 3], [2, 8]],
    [[3, 9], [4, 2]]
]]]]);
  
// Print the tensor
example1.print()

输出：

Tensor
    [[[[[[1, 3],
         [2, 8]],

        [[3, 9],
         [4, 2]]]]]]

示例 2：在此示例中，我们正在创建张量，我们将在其中传递平面数组并指定张量的形状参数。我们将在这里看到 shape 参数的用法。

Javascript

// Import the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Define the value of the tensor
var value = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12];
  
// Specify the shape of the tensor
var shape = [1, 2, 6, 1, 1, 1];
  
// Create the tensor
var example2 = tf.tensor6d(value, shape);
  
// Print the tensor
example2.print();

输出：

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

      [ [ [[7 ],]],
        [ [[8 ],]],
        [ [[9 ],]],
        [ [[10],]],
        [ [[11],]],
        [ [[12],]]]]]

示例 3：在此示例中，我们将通过指定值、形状和数据类型来创建张量。我们将创建所有值都是字符串数据类型的张量。

Javascript

// Import the tensorflow.js library
import * as tf from "@tensorflow/tfjs";
  
// Define the value of the tensor
var value = ["C", "C++", "Java", "Python",
            "PHP", "JS", "SQL", "React"];
  
// Specify the shape of the tensor
var shape = [1, 2, 4, 1, 1, 1];
  
// Create the tensor
var example3 = tf.tensor6d(value, shape);
  
// Print the tensor
example3.print();

输出：

Tensor
    [[[ [ [['C'     ],]],
        [ [['C++'   ],]],
        [ [['Java'  ],]],
        [ [['Python'],]]],

      [ [ [['PHP'   ],]],
        [ [['JS'    ],]],
        [ [['SQL'   ],]],
        [ [['React' ],]]]]]

参考： https://js.tensorflow.org/api/latest/#tensor6d