tf.TensorSpec
| Просмотреть исходный код на GitHub |
Описывает tf.Tensor.
Наследуется от: TypeSpec, TraceType
tf.TensorSpec(
shape,
dtype=tf.dtypes.float32,
name=None
)
Метаданные для описания объектов tf.Tensor, принимаемых или возвращаемых некоторыми API TensorFlow.
| Аргументы | |
|---|---|
shape | Значение, преобразуемое в tf.TensorShape. Форма тензора. |
dtype | Значение, преобразуемое в tf.DType. Тип значений тензора. |
name | Необязательное имя для тензора. |
| Исключения | |
|---|---|
TypeError | Если форма не преобразуется в tf.TensorShape, или тип данных не преобразуется в tf.DType. |
| Атрибуты | |
|---|---|
dtype | Возвращает количество элементов в тензоре. |
name | Возвращает (необязательно предоставленное) имя описанного тензора. |
shape | Возвращает TensorShape, представляющий форму тензора. |
value_type | Тип Python для значений, совместимых с этим TypeSpec. |
Методы
from_spec
@classmethod
from_spec(
spec, name=None
)
Возвращает тензор с той же формой и типом данных, что и 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')
| Аргументы | |
|---|---|
spec | Спецификация, используемая для создания нового тензора. |
name | Имя для нового тензора. По умолчанию spec.name. |
from_tensor
@classmethod
from_tensor(
tensor, name=None
)
Возвращает спецификацию тензора, описывающую tensor.
tf.TensorSpec.from_tensor(tf.constant([1, 2, 3])) TensorSpec(shape=(3,), dtype=tf.int32, name=None)
| Аргументы | |
|---|---|
tensor | tf.Tensor, который должен быть описан. |
name | Имя для спецификации тензора. По умолчанию tensor.op.name. |
| Возвращает | |
|---|---|
Спецификация тензора, описывающая tensor. |
is_compatible_with
is_compatible_with(
spec_or_tensor
)
Возвращает True, если spec_or_tensor совместим с этим TensorSpec.
Два тензора считаются совместимыми, если у них одинаковый тип данных и формы совместимы (см. tf.TensorShape.is_compatible_with).
| Аргументы | |
|---|---|
spec_or_tensor | tf.TensorSpec или tf.Tensor |
| Возвращает | |
|---|---|
| True, если spec_or_tensor совместим с self. |
is_subtype_of
is_subtype_of(
other: tf.types.experimental.TraceType
) -> bool
Возвращает True, если self является подтипом other.
Реализует интерфейс tf.types.experimental.func.TraceType.
Если не переопределено подклассом, по умолчанию предполагается, что TypeSpec является ковариативным по атрибутам, реализующим TraceType, и инвариантным по остальным атрибутам, а также по структуре и типу TypeSpec.
| Аргументы | |
|---|---|
other | Объект TraceType. |
most_specific_common_supertype
most_specific_common_supertype(
others: Sequence[tf.types.experimental.TraceType]
) -> Optional['TypeSpec']
Возвращает наиболее специфичный тип-предок TypeSpec для self и others.
Реализует интерфейс tf.types.experimental.func.TraceType.
Если не переопределено подклассом, по умолчанию предполагается, что TypeSpec является ковариативным по атрибутам, реализующим TraceType, и инвариантным по остальным атрибутам, а также по структуре и типу TypeSpec.
| Аргументы | |
|---|---|
others | Последовательность TraceTypes. |
most_specific_compatible_type
most_specific_compatible_type(
other: 'TypeSpec'
) -> 'TypeSpec'
Возвращает наиболее специфичную TypeSpec, совместимую с self и other. (устарело)
Устарело. Используйте, пожалуйста, most_specific_common_supertype вместо этого. Не переопределяйте эту функцию.
| Аргументы | |
|---|---|
other | TypeSpec. |
| Исключения | |
|---|---|
ValueError | Если нет TypeSpec, совместимого с self и other. |
__eq__
__eq__(
other
)
Возвращает self==value.
__ne__
__ne__(
other
)
Возвращает self!=value.
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/TensorSpec