tf.TensorSpec

TensorFlow 1 version View source on GitHub

Describes a tf.Tensor.

tf.TensorSpec(
    shape, dtype=tf.dtypes.float32, name=None
)

Used in the notebooks

Used in the guide Used in the tutorials

Metadata for describing the tf.Tensor objects accepted or returned by some TensorFlow APIs.

Args:

  • shape: Value convertible to tf.TensorShape. The shape of the tensor.
  • dtype: Value convertible to tf.DType. The type of the tensor values.
  • name: Optional name for the Tensor.

Attributes:

  • dtype: Returns the dtype of elements in the tensor.
  • name: Returns the (optionally provided) name of the described tensor.
  • shape: Returns the TensorShape that represents the shape of the tensor.
  • value_type

Raises:

Methods

__eq__

View source

__eq__(
    other
)

Return self==value.

__ne__

View source

__ne__(
    other
)

Return self!=value.

from_spec

View source

@classmethod
from_spec(
    spec, name=None
)

from_tensor

View source

@classmethod
from_tensor(
    tensor, name=None
)

is_compatible_with

View source

is_compatible_with(
    spec_or_tensor
)

Returns True if spec_or_tensor is compatible with this TensorSpec.

Two tensors are considered compatible if they have the same dtype and their shapes are compatible (see tf.TensorShape.is_compatible_with).

Args:

  • spec_or_tensor: A tf.TensorSpec or a tf.Tensor

Returns:

True if spec_or_tensor is compatible with self.

most_specific_compatible_type

View source

most_specific_compatible_type(
    other
)

Returns the most specific TypeSpec compatible with self and other.

Args:

  • other: A TypeSpec.

Raises:

  • ValueError: If there is no TypeSpec that is compatible with both self and other.