tf.raw_ops.RestoreSlice
Восстанавливает тензор из файлов контрольной точки.
tf.raw_ops.RestoreSlice(
file_pattern,
tensor_name,
shape_and_slice,
dt,
preferred_shard=-1,
name=None
)
Это похоже на Restore, за исключением того, что восстановленный тензор может быть указан как заполняющий только срез более крупного тензора. shape_and_slice задает форму более крупного тензора и срез, который занимает восстановленный тензор.
Вход shape_and_slice имеет тот же формат, что и элементы входа shapes_and_slices операции SaveSlices.
| Аргументы | |
|---|---|
file_pattern | Тензор Tensor типа string. Должен содержать единственный элемент. Шаблон файлов, из которых считывается тензор. |
tensor_name | Тензор Tensor типа string. Должен содержать единственный элемент. Имя тензора, который нужно восстановить. |
shape_and_slice | Тензор Tensor типа string. Скаляр. Формы и спецификации срезов, используемые при восстановлении тензора. |
dt | Объект tf.DType. Тип восстанавливаемого тензора. |
preferred_shard | Необязательный int. По умолчанию — -1. Индекс файла, который нужно открыть первым, если соответствует несколько файлов: file_pattern. См. документацию по Restore. |
name | Имя операции (необязательно). |
| Возвращает | |
|---|---|
Тензор Tensor типа dt. |
© 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/raw_ops/RestoreSlice
