假设我有一个形状（大小，大小）的2D张量，我想得到2个包含原始张量行索引和列索引的新张量 . 所以如果size是2，我想得到[[0,0]，[1,1]]和[[0,1]，[0,1]]
棘手的是，size是另一个张量，只有在张量流Session中运行图形时才能知道它的值 . 我怎么能在张量流中做到这一点？
好像你在寻找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 ），这可以略微简化 .
matrix
R
C
tf.range
1 回答
好像你在寻找tf.meshgrid .
这是一个例子：
matrix是您的2D张量，R和C分别包含您的行索引和列索引 . 请注意，如果矩阵是正方形（只有一个tf.range），这可以略微简化 .