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 ofxy
orij
.
例如:
用张量调用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