阅读(2915) (11)

TensorFlow张量变换:tf.meshgrid

2017-11-02 15:59:13 更新
tf.meshgrid 函数
meshgrid(
    *args,
    **kwargs
)

定义在:tensorflow/python/ops/array_ops.py.

参见指南:张量变换>张量形状的确定与改变

广播用于计算 N 维网格的参数.

给定 N 个一维坐标数组 *args,返回 N 维坐标数组的列表输出,用于计算 N 维网格上的表达式.

Note:meshgrid 支持 cartesian ('xy') 和 matrix ('ij') 索引约定.当 indexing 参数设置为 “xy”(默认值)时,前两个维度的广播指令被交换.

例子:

调用 X, Y = meshgrid(x, y),其具有如下张量:

x = [1, 2, 3]
y = [4, 5, 6]

结果是:

X = [[1, 2, 3],
     [1, 2, 3],
     [1, 2, 3]]
Y = [[4, 4, 4],
     [5, 5, 5],
     [6, 6, 6]]

参数:

  • args:秩为1的张量.
  • indexing 1.:“xy”或“ij”(可选值,默认值为:'xy').
  • * name:操作的名称(可选).

返回:

  • outputs:秩为 N 的 N 张量列表.