tf.train.experimental.ShardableTensor
Обёртка над тензором, содержащая данные, необходимые для фрагментации.
tf.train.experimental.ShardableTensor(
_tensor_save_spec: saveable_object.SaveSpec,
tensor: tf.Tensor,
dtype: tf.dtypes.DType,
device: tf.DeviceSpec,
name: str,
shape: tf.TensorShape,
slice_spec: tf.Variable.SaveSliceInfo,
checkpoint_key: str,
trackable: base.Trackable
)
Представление тензора, используемое в качестве входных данных для готовых и пользовательских tf.train.experiemental.ShardingCallback, которые можно указать с помощью опции experimental_sharding_callback в tf.train.CheckpointOptions.
| Атрибуты | |
|---|---|
tensor | Поле класса данных |
dtype | Поле класса данных |
device | Поле класса данных |
name | Поле класса данных |
shape | Поле класса данных |
slice_spec | Поле класса данных |
checkpoint_key | Поле класса данных |
trackable | Поле класса данных |
Методы
__eq__
__eq__(
other
)
© 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/api_docs/python/tf/train/experimental/ShardableTensor
