Python tf.get_static_value用法及代码示例

如果可有效计算，则返回给定张量的常量值。

用法

tf.get_static_value(
    tensor, partial=False
)

此函数尝试对给定的张量进行部分评估，如果成功，则将其值作为 numpy ndarray 返回。

示例用法：

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

在 tf.function 中调用 get_static_value 时，使用 partial 选项最为相关。将其设置为 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 提供不同的值。这允许此函数的结果影响构建的图形，并允许静态形状优化。