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

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

tf.depthToSpace()是tensorflow.js库的内置函数，用于重新排列输入张量中的数据，其中深度维度中的值在空间块中移动到高度维度和宽度维度。它将数据从深度重新排列为空间数据块。

用法:

tensor.depthToSpace(input, blocksize, dataformat)

参数表：

input:给定张量
blocksize:输出张量的宽度为depth * blockSize。
dataformat：指定给定和结果张量的布局。它有两个选项：“NHWC”：[批处理，高度，宽度，通道]和“NCHW”：[批处理，通道，高度，宽度]

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

范例1：使用“NHWC”格式

Javascript


// Importing the tensorflow.Js lbrary
import * as tf from "@tensorflow/tfjs"
  
// Create a new tensor
var input = tf.tensor4d([1, 3, 5, 7], [1, 1, 1, 4]);
  
// define block size
var blockSize = 2;
  
// define data format
var dataFormat = "NHWC";
  
// rearrange data
var val = tf.depthToSpace(input, blockSize, dataFormat);
  
// print the tensor
val.print();

输出

Tensor
    [[[[1],
       [3]],

      [[5],
       [7]]]]

范例2：使用“NCHW”格式

Javascript


// Importing the tensorflow.Js lbrary
import * as tf from "@tensorflow/tfjs"
  
// Create a new tensor
var input = tf.tensor4d([1, 3, 5, 7], [1, 4, 1, 1]);
  
// define block size
var blockSize = 2;
  
// define data format
var dataFormat = "NCHW";
  
// rearrange data
var tr = tf.depthToSpace(input, blockSize, dataFormat);
  
// print the tensor
tr.print();

输出

Tensor
    [[[[1, 3],
       [5, 7]]]]

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

相关用法

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