tf.meshgrid

参考  tf.meshgrid - 云+社区 - 腾讯云

tf.meshgrid(
    *args,
    **kwargs
)

在N-D网格上广播用于评估的参数。给定N个一维坐标数组*args，返回一个N- d坐标数组的列表输出，用于计算N- d网格上的表达式。

注意:

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

参数:

• *args: 秩为1的张量。
• **kwargs: -索引:  要么'xy'或'ij'(可选，默认为'xy')。
• name: 操作的名称(可选)。

返回值:

• outputs:  N个秩为N的张量的列表。

可能产生的异常:

• TypeError: When no keyword arguments (kwargs) are passed.
• ValueError: When indexing keyword argument is not one of xy or ij.

例如:

用张量调用X, Y = meshgrid(X, Y)

import tensorflow as tf

a=[0,5,10]
b=[0,5,15,20,25]
A,B=tf.meshgrid(a,b)
with tf.Session() as sess:
  print (A.eval())
  print (B.eval())



Output:
-------------
[[ 0  5 10]
 [ 0  5 10]
 [ 0  5 10]
 [ 0  5 10]
 [ 0  5 10]]
[[ 0  0  0]
 [ 5  5  5]
 [15 15 15]
 [20 20 20]
 [25 25 25]]
-------------

注意：两个矩阵的维数相同，用b的维数作为行，用a的维数作为列。

原链接:  https://tensorflow.google.cn/versions/r1.9/api_docs/python/tf/meshgrid?hl=en