Python tf.type_spec_from_value用法及代码示例

返回表示给定 value 的 tf.TypeSpec 。

用法

tf.type_spec_from_value(
    value
) -> tf.TypeSpec

参数

value TensorFlow API 可以接受或返回的值。 value 接受的类型包括 tf.Tensor 、可以使用 tf.convert_to_tensor 转换为 tf.Tensor 的任何值，以及 CompositeTensor 的任何子类(例如 tf.RaggedTensor )。

与 value 兼容的 TypeSpec 。

抛出

TypeError 如果无法为 value 构建 TypeSpec，因为它的类型不受支持。

例子：

tf.type_spec_from_value(tf.constant([1, 2, 3]))
TensorSpec(shape=(3,), dtype=tf.int32, name=None)
tf.type_spec_from_value(np.array([4.0, 5.0], np.float64))
TensorSpec(shape=(2,), dtype=tf.float64, name=None)
tf.type_spec_from_value(tf.ragged.constant([[1, 2], [3, 4, 5]]))
RaggedTensorSpec(TensorShape([2, None]), tf.int32, 1, tf.int64)

example_input = tf.ragged.constant([[1, 2], [3]])
@tf.function(input_signature=[tf.type_spec_from_value(example_input)])
def f(x):
  return tf.reduce_sum(x, axis=1)

相关用法

注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.type_spec_from_value。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。