في نظرية الأعداد, تكون دالة الوحدة (بالإنجليزية: Unit function) دالة جداءية بشكل كامل على الأعداد الصحيحة الموجبة المعرفة كالتالي:
تسمى دالة الوحدة لأنها العنصر المحايد لالتواء ديريشليه.
يمكن أن توصف هذه الدالة باسم «دالة المؤشر للعنصر 1» ضمن مجموعة الأعداد الصحيحة. ويمكن أن تُكتب بهذه الطريقة (لاحظ بإنها مختلفة عن ).