如何將張量映射到它在張量流中的指數

Question

假設我有一個具有形狀（大小，尺寸）的二維張量，並且我想得到2個包含原始張量行索引和列索引的新張量。因此，如果大小爲2，我想獲得 [[0,0]，[1,1]]和[[0,1]，[0，1]]

棘手的是，尺寸是另一個張量的值只有在張量流會話中運行時才能知道。 我如何在tensorflow中做到這一點？

Answer 1

好像您在尋找tf.meshgrid。

下面是一個例子：

shape = tf.shape(matrix) 
R, C = tf.meshgrid(tf.range(shape[0]), tf.range(shape[1]), indexing='ij') 

matrix是你的2D張量，R和C包含您的行和列索引，分別。請注意，如果您的矩陣是方形的（只有一個tf.range），這可以稍微簡化。