TensorFlow張量變換:tf.meshgrid

2018-11-02 16:20 更新
tf.meshgrid 函數(shù)
meshgrid(
    *args,
    **kwargs
)

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

參見指南:張量變換>張量形狀的確定與改變

廣播用于計(jì)算 N 維網(wǎng)格的參數(shù).

給定 N 個(gè)一維坐標(biāo)數(shù)組 *args,返回 N 維坐標(biāo)數(shù)組的列表輸出,用于計(jì)算 N 維網(wǎng)格上的表達(dá)式.

Note:meshgrid 支持 cartesian ('xy') 和 matrix ('ij') 索引約定.當(dāng) indexing 參數(shù)設(shè)置為 “xy”(默認(rèn)值)時(shí),前兩個(gè)維度的廣播指令被交換.

例子:

調(diào)用 X, Y = meshgrid(x, y),其具有如下張量:

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

結(jié)果是:

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

參數(shù):

  • args:秩為1的張量.
  • indexing 1.:“xy”或“ij”(可選值,默認(rèn)值為:'xy').
  • * name:操作的名稱(可選).

返回:

  • outputs:秩為 N 的 N 張量列表.
以上內(nèi)容是否對您有幫助:
在線筆記
App下載
App下載

掃描二維碼

下載編程獅App

公眾號
微信公眾號

編程獅公眾號