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