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.ShardingCallbacks, которые можно задать с помощью параметра experimental_sharding_callback в tf.train.CheckpointOptions.

Методы

__eq__

__eq__(
    other
)