sponge.function.calc_distance
- sponge.function.calc_distance(position_a: Tensor, position_b: Tensor, pbc_box: Tensor = None, keepdims: bool = False)[source]
Compute distance between position
and .- Parameters
position_a (Tensor) – Tensor of shape
. Data type is float. Position coordinate of point .position_b (Tensor) – Tensor of shape
. Data type is float. Position coordinate of point .pbc_box (Tensor) – Tensor of shape
or . Data type is float. Size of PBC boxkeepdims (bool) – If this is set to
True
, the last axis will be left in the result as dimensions with size one. Default:False
.
- Returns
distance (Tensor), a tensor of shape
or . Data type is float.
- Supported Platforms:
Ascend
GPU