tf.range()乘法的优雅方法

Question

我有一个(BatchSize X 1)数的张量，介于0到9之间，例如：

3
4
2
.
.
.
2

从每个数字开始，我需要创建一个新的张量，从数字开始，直到数字- 9。例如，对于数字3，我要创建一个张量：

3 2 1 0 -1 -2 -3 -4 -5 -6

这可以使用一个单独的：

tf.range(3,3-9,1)

我需要对整个张量执行此操作，以创建一个新的张量(遵循上面的示例)：

3 2 1 0 -1 -2 -3 -4 -5 -6
4 3 2 1 0 -1 -2 -3 -4 -5
2 1 0 -1 -2 -3 -4 -5 -6 -7
. . . . . . . . . .
. . . . . . . . . .
. . . . . . . . . .
2 1 0 -1 -2 -3 -4 -5 -6 -7

一种选择是以迭代的方式完成，但这需要我将批处理大小硬编码到循环中，我正在努力避免这种情况。

任何好的优雅解决方案的想法都将受到欢迎。

Answer 1

您可以使用tf.range(10)对张量进行外部减法，方法是对张量进行适当的整形：

import tensorflow as tf
tf.InteractiveSession()
a = tf.constant([3,4,2,2])

b = tf.reshape(a, (-1,1)) - tf.range(10)
b.eval()
#array([[ 3,  2,  1,  0, -1, -2, -3, -4, -5, -6],
#       [ 4,  3,  2,  1,  0, -1, -2, -3, -4, -5],
#       [ 2,  1,  0, -1, -2, -3, -4, -5, -6, -7],
#       [ 2,  1,  0, -1, -2, -3, -4, -5, -6, -7]], dtype=int32)