sponge.function.pbc_box_reshape

View Source On Gitee
sponge.function.pbc_box_reshape(pbc_box: Tensor, ndim: int)[source]

Reshape the pbc_box as the same ndim.

Parameters

  • pbc_box (Tensor) – Tensor of shape \((B, D)\). Data type is float. B is batchsize, i.e. number of walkers in simulation. D is spatial dimension of the simulation system. Usually is 3.

  • ndim (int) – The rank (number of dimension) of the pbc_box

Returns

pbc_box (Tensor), a tensor of shape \((B, 1, .., 1, D)\). Data type is float.

Supported Platforms:

Ascend GPU