multidim-positional-encoding - 支持PyTorch和TensorFlow的1D、2D和3D正弦位置编码实现
Github开源项目PyTorchTensorFlowPositional Encoding1D Encoding2D Encoding
这是一款易于下载和实用的1D、2D和3D正弦位置编码工具,适用于PyTorch和TensorFlow。支持处理(batchsize, x, ch)、(batchsize, x, y, ch)和(batchsize, x, y, z, ch)形式的张量,并在ch维度进行编码。该工具扩展了一维位置编码至多维,提供了详细的安装和使用指南,并在6.0.1版本后优化了代码导入方式。