W3cubDocs

/TensorFlow 2.4

tf.type_spec_from_value

Returns a tf.TypeSpec that represents the given value.

Examples:

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)
Args
value A value that can be accepted or returned by TensorFlow APIs. Accepted types for value include tf.Tensor, any value that can be converted to tf.Tensor using tf.convert_to_tensor, and any subclass of CompositeTensor (such as tf.RaggedTensor).
Returns
A TypeSpec that is compatible with value.
Raises
TypeError If a TypeSpec cannot be built for value, because its type is not supported.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/type_spec_from_value