View source on GitHub |
Describes a tf.Tensor.
Inherits From: TypeSpec
tf.TensorSpec(
shape,
dtype=tf.dtypes.float32
,
name=None
)
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. |
Raises | |
---|---|
TypeError
|
If shape is not convertible to a tf.TensorShape , or dtype is
not convertible to a tf.DType .
|
Methods
from_spec
@classmethod
from_spec( spec, name=None )
Returns a TensorSpec
with the same shape and dtype as spec
.
spec = tf.TensorSpec(shape=[8, 3], dtype=tf.int32, name="OriginalName")
tf.TensorSpec.from_spec(spec, "NewName")
TensorSpec(shape=(8, 3), dtype=tf.int32, name='NewName')
Args | |
---|---|
spec
|
The TypeSpec used to create the new TensorSpec .
|
name
|
The name for the new TensorSpec . Defaults to spec.name .
|
from_tensor
@classmethod
from_tensor( tensor, name=None )
Returns a TensorSpec
that describes tensor
.
tf.TensorSpec.from_tensor(tf.constant([1, 2, 3]))
TensorSpec(shape=(3,), dtype=tf.int32, name=None)
Args | |
---|---|
tensor
|
The tf.Tensor that should be described.
|
name
|
A name for the TensorSpec . Defaults to tensor.op.name .
|
Returns | |
---|---|
A TensorSpec that describes tensor .
|
is_compatible_with
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
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 .
|
__eq__
__eq__(
other
)
Return self==value.
__ne__
__ne__(
other
)
Return self!=value.