tf.experimental.dtensor.name_based_save

Сохраняет тензор с именем в контрольной точке.

tf.experimental.dtensor.name_based_save(
    mesh: tf.experimental.dtensor.Mesh,
    checkpoint_prefix: Union[str, tf.Tensor],
    name_tensor_dict: Dict[str, Union[tensor_lib.Tensor, tf_variables.Variable]]
)

Функция подготавливает входной словарь к формату sharded_save, чтобы воспользоваться преимуществами распределённого сохранения на основе SPMD DTensor.

Так же, как и при восстановлении, функция поддерживает сохранение только на одной сетке.