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

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

.stridedSlice() 函数用于拉出指定输入张量的跨步部分。

注意：从指定的输入张量开始。从 begin 给定的位置开始，切片通过在所有测量值大于 end 之前将步幅附加到规定的索引来进行。此外，步幅也可能是负数，这会导致反向切片。

用法:

tf.stridedSlice(x, begin, end, strides?, beginMask?, endMask?, 
ellipsisMask?, newAxisMask?, shrinkAxisMask?)

Parameters:

x:规定的张量以跨过切片。它可以是 tf.Tensor、TypedArray 或 Array 类型。
begin:切片开始的指定坐标。它的类型为 number[]。
end:切片结束的指定坐标。它的类型为 number[]。
strides:切片的规定长度。它是可选的，类型为 number[]。
beginMask:它是一个可选参数，类型为 number。如果 beginMask 的第 i 位是固定的，则忽略 begin[i] 并且相反地利用该维度中的最大可达到范围。
endMask:它是一个可选参数，类型为 number。如果 endMask 的第 i 位是固定的，则忽略 end[i] 并且相反地利用该维度中的最大可达到范围。
ellipsisMask:它是类型为 number 的可选参数。
newAxisMask:它是类型为 number 的可选参数。
shrinkAxisMask:指定的位掩码，其中位 i 表示第 i 个指定必须缩小容量。开始和结束应该指示长度为 1 的切片。它是可选的，类型为 number。

返回值：它返回tf.Tensor。

范例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining tensor input
const tn = tf.tensor3d([5, 5, 5 ,7, 7, 7, 13, 
    13, 13, 14, 14, 14, 1, 1, 1, 2, 2, 2],
    [3, 3, 2]);
  
// Calling stridedSlice() method and 
// Printing output
tn.stridedSlice([5, 0, 5], [7, 5, 7], [2, 2, 2]).print();

输出：

Tensor
    [[[1],
      [2]]]

范例2：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining tensor input
const tn = tf.tensor3d([5, 5, 5 ,7, 7, 7, 13, 
    13, 13, 14, 14, 14, 1, 1, 1, 2, 2, 2],
    [3, 3, 2]);
  
// Defining all the parameters
const x = [-5, 5, 7];
const begin = [-7, 13, 14];
const end = [-7, 13, 13];
const strides = [1];
const beginMask = 2;
const endMask = 0;
const ellipsisMask = 2;
const shrinkAxisMask = 15;
  
// Calling stridedSlice() method and 
// Printing output
tn.stridedSlice(x, begin, end, strides, beginMask,
     endMask, ellipsisMask, shrinkAxisMask).print();

输出：

Tensor
    2

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

相关用法

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