分子模拟简介
使用者指南
API参考
RELEASE NOTES
根据指标从张量的倒数第二轴收集向量。
tensor (Tensor) - 输入张量,shape为 \((B, X, D)\) 。其中 \(B\) 为batch size, \(X\) 为任意大小, \(D\) 为模拟系统的维度,通常为3。
index (Tensor) - 索引,shape为 \((B, ...,)\)。
Tensor。取出的向量。shape为 \((B, ..., D)\)。
Ascend GPU
Ascend
GPU