基于逐元素运算的图像处理方法、装置、设备及介质

本申请提供了一种基于逐元素运算的图像处理方法、装置、设备及介质，涉及GPU技术领域，该方法包括：根据需要参与运算的多个所述输入张量中包含的多个元素的元素个数，以及GPU设备的硬件资源信息，确定所述多个输入张量在逐元素运算过程中需要使用的线程块数量；根据所述线程块数量、所述元素个数以及所述多个元素的内存布局，基于预先定义的模板结构体，确定所述需要参与运算的多个输入张量对应的逐元素运算类算子的内核函数，所述模板结构体的参数包含元素类型和元素个数；基于所述线程块数量，发起多个线程，以使所述多个线程，按照所述内核函数，获取所述多个元素，并执行所述多个输入张量的逐元素运算过程，得到图像处理结果。