17.6. Random seeds

popxl.create_seeds(seed, offset=0, batches_per_step=1, gradient_accumulation_factor=1, replicas=1)

Create seed tensors from a parent seed.

A seed tensor is a uint64 value represented as two uint32 elements. The function creates a NumPy array containing batches_per_step * gradient_accumulation_factor * replicas seed tensors. The result is a tensor with shape [batches_per_step, gradient_accumulation_factor, replicas, 2].

Parameters

  • seed (int) – Seed value for the random number generator.

  • offset (int) – Offset added to seed.

  • batches_per_step (int, optional) – Number of batches per step. Defaults to 1.

  • replicas (int, optional) – Number of model replications. Defaults to 1.

  • gradient_accumulation_factor (int, optional) – Gradient accumulation factor of model. Defaults to 1.

Returns

The initialised seed tensors.

Return type

numpy.ndarray