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 提供不同的值。這允許此函數的結果影響構建的圖形，並允許靜態形狀優化。