Tensorflow.js tf.meshgrid()函数

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

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

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

句法：

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

参数：

x：所述张量以及等级 geq 为一。它是可选的，可以是 tf.Tensor、TypedArray 或 Array 类型。
y：所述张量以及等级 geq 为一。它是可选的，可以是 tf.Tensor、TypedArray 或 Array 类型。
__2:可选参数，类型为 { indexing?: 字符串; }。

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

示例 1：使用秩为 1 的张量。

Javascript

// 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);

Javascript

// 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
    [[1, 1, 3],
     [1, 1, 3],
     [1, 1, 3]],Tensor
    [[2, 2, 2],
     [5, 5, 5],
     [4, 4, 4]]

示例 2：使用浮点值。

Javascript

// 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