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

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

.meshgrid() 函数用于广播参数，以便在 N-D 网格上进行分析。此外，对于一维坐标数组，即 *args，此方法返回 N-D 坐标数组的输出列表，用于分析 N-D 网格上给定 N 的表达式。

注意：此函数支持笛卡尔 ‘xy’ 以及矩阵 ‘ij’ 索引协议。如果索引参数固定为默认值，即 ‘xy’，则交换前两个测量的广播命令。

用法:

tf.meshgrid(x?, y?, __2?)

Parameters: 

• x:规定的张量以及排名 geq 一。它是可选的，可以是 tf.Tensor、TypedArray 或 Array 类型。
• y:规定的张量以及排名 geq 一。它是可选的，可以是 tf.Tensor、TypedArray 或 Array 类型。
• __2:它是可选参数，类型为 { indexing?:string; }.

返回值：它返回tf.Tensor []。

范例1：使用等级 1 的张量。

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining first tensor
const t1 = [1, 1, 3];
  
// Defining second tensor
const t2 = [2, 5, 4];
  
// Calling meshgrid() function
const res = tf.meshgrid(t1, t2);
  
// Printing output
console.log(res);

输出：

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

范例2：使用浮点值。

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Calling meshgrid() function with
// float values
const output = tf.meshgrid(2.1, 3.3);
  
// Printing output
console.log(output);

输出：

Tensor
     [[2.0999999],],Tensor
     [[3.3],]

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