在TPU芯片上的数据处理方法、装置、设备及介质

本发明提供在TPU芯片上的数据处理方法、装置、设备及介质，涉及伪随机数生成技术领域。本发明能够通过将各计数器数据连续存储在TPU内存中，基于计数器内存地址确定计数器偏移量，基于偏移量确定对应的张量核心的编号。根据计数器偏移量和张量核心编号得到各不相同的计数器值，确保伪随机数的唯一性。连续存储的计数器值可基于张量核心的最大处理能力为单位，批量加载至向量寄存器。各计数器映射至各张量核心。基于不同的计数器值，每个张量核心采用点对点方式独立处理分配的计数器数据，并行生成多个不同的伪随机数。本发明充分利用TPU芯片的并行处理能力，并行批量生成伪随机数，提升了在TPU芯片上生成伪随机数的效率。