sponge.function.pbc_box_reshape

sponge.function.pbc_box_reshape(pbc_box: Tensor, ndim: int)

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