Question

假设我有一个形状（大小，大小）的2D张量，我想得到2个包含原始张量行索引和列索引的新张量。所以，如果大小是2，我想得到 [[0,0]，[1,1]]和[[0,1]，[0,1]]

棘手的是，size是另一个张量，只有在张量流会话中运行图形时才能知道它的值。我怎样才能在张量流中做到这一点？

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），则可以略微简化。