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

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

tf.reshape() 函数用于用指定的形状重塑给定的张量。

用法:

tf.reshape(x, shape)

参数：该函数具有以下参数：

x:它是需要整形的输入张量。
shape:我们需要传递数字数组来定义输出形状。

返回值：它返回一个tf.Tensor。

范例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
const x = tf.tensor1d([10, 15, 16, 24]);
  
// Print the tensor
x.reshape([2, 2]).print();

输出：

Tensor
    [[10, 15],
     [16, 24]]

范例2：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Using 2d
const x = tf.tensor2d(
  [1, 2, 3, 4, 5, 6, 7, 8, 9], [3, 3]
);
x.reshape([3, 3]).print();
  
// Using 3d
const y = tf.tensor3d(
  [[[1], [2]], [[3], [4]]]
);
y.reshape([2, 2]).print();
  
// Using 4d
const z = tf.tensor4d(
  [11, 12, 13, 14], [1, 2, 2, 1]
);
z.reshape([2, 2]).print();

输出：

Tensor
    [[1, 2, 3],
     [4, 5, 6],
     [7, 8, 9]]
Tensor
    [[1, 2],
     [3, 4]]
Tensor
    [[11, 12],
     [13, 14]]

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

相关用法

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