I mean the enemy AI will always try to save a unit before death by making it burrow if it can, which can be annoying if you don't have detection. If it's player controlled, no problem.
And the unit must have an animation for that of course and there aren't that many : crypt fiends, scarabs, some custom models here on hive... What unit do you want ?