Python tf.meshgrid用法及代码示例

广播参数以在N-D 网格上进行评估。

用法

tf.meshgrid(
    *args, **kwargs
)

给定 N 个一维坐标数组 *args ，返回 N-D 坐标数组的列表 outputs，用于评估 N-D 网格上的表达式。

注意:

meshgrid 支持笛卡尔 ('xy') 和矩阵 ('ij') 索引约定。当indexing 参数设置为'xy'(默认值)时，交换前两个维度的广播指令。

例子：

使用张量调用X, Y = meshgrid(x, y)

x = [1, 2, 3]
y = [4, 5, 6]
X, Y = tf.meshgrid(x, y)
# X = [[1, 2, 3],
#      [1, 2, 3],
#      [1, 2, 3]]
# Y = [[4, 4, 4],
#      [5, 5, 5],
#      [6, 6, 6]]