x = tf.reshape(tf.range(30, dtype=tf.float32), [10, 3])
with tf.GradientTape() as g:
g.watch(x)
y = tf.gather(x, [1, 2]) # IndexedSlices
z = tf.math.scalar_mul(10.0, y)
Аргументы
scalar
Скалярное значение 0-й размерности Tensor. Размерность должна быть известна.
x
Объект Tensor или IndexedSlices для масштабирования.
name
Имя операции (необязательно).
Возвращаемое значение
Результат умножения, того же типа (Tensor или IndexedSlices) что и x.