tf.get_static_value

Возвращает постоянное значение заданного тензора, если его можно эффективно вычислить.

tf.get_static_value(
    tensor, partial=False
)

Эта функция пытается частично оценить заданный тензор и возвращает его значение в виде массива numpy, если это удается.

Пример использования:

a = tf.constant(10)
tf.get_static_value(a)
10
b = tf.constant(20)
tf.get_static_value(tf.add(a, b))
30

# `tf.Variable` is not supported.
c = tf.Variable(30)
print(tf.get_static_value(c))
None

Использование partial опции наиболее актуально при вызове get_static_value внутри tf.function. Установка ее в True вернёт результаты, но для значений, которые нельзя оценить, будут None. Например:

class Foo(object):
  def __init__(self):
    self.a = tf.Variable(1)
    self.b = tf.constant(2)

  @tf.function
  def bar(self, partial):
    packed = tf.raw_ops.Pack(values=[self.a, self.b])
    static_val = tf.get_static_value(packed, partial=partial)
    tf.print(static_val)

f = Foo()
f.bar(partial=True)  # `array([None, array(2, dtype=int32)], dtype=object)`
f.bar(partial=False)  # `None`

Совместимость (V1): Если constant_value(tensor) возвращает результат, отличный от None, то уже не будет возможности передать другое значение для tensor. Это позволяет результату этой функции влиять на создаваемую график и позволяет выполнять оптимизацию статической формы.