sponge.partition.IndexVectors

class sponge.partition.IndexVectors(use_pbc: bool = None)

按索引获取向量。

参数：

• use_pbc (bool，可选) - 是否使用周期性边界条件。默认值： None，默认取 False。。

支持平台：

Ascend GPU

样例：

>>> import sponge
>>> from sponge.partition import IndexVectors
>>> import mindspore
>>> from mindspore import Tensor
>>> idx_vector = IndexVectors(use_pbc=False)
>>> coordinate = Tensor([[[0.0, 0.0, 0.0], [0.0, 0.0, 1.0]]])
>>> index = Tensor([[[1],[0]]])
>>> print(idx_vector(coordinate, index))  # B=1, A=2, N=1
[[[[ 0.  0.  1.]]
[[ 0.  0. -1.]]]]