mindspore.ops.MDIterationSetupRandState

class mindspore.ops.MDIterationSetupRandState(atom_numbers, seed)

Compute the random state of the iteration.

Because there is a large amount of inputs and each of them are related, there is no way to construct Examples using random methods. For details, refer the webpage SPONGE in MindSpore.

Parameters

• atom_numbers (int32) – the number of atoms n.

• seed (int32) – random seed.

Outputs:

• output (Tensor) random state. The data type is float32 and the shape is \((ceil(n * 3 / 4),)\).

Supported Platforms:

GPU