How to make this function to work also for scalars, if possible?

In this way:

But right now I inserted manually 1x1 matrix element.
So, giving a scalar value like (or any scalar value):

expm function to be able to transform this scalar into 1x1 matrix element. Is it possible like this to add something like that in the above expm function? Or I need to be careful and put manually the 1x1 matrix element like above when I want to perform a calculation with this expm function with scalar values?

In (express) Prime your function could be:

then:

gives:

and:

gives:

Success!
Luc

if A is a scalar, simply turn it into a 1x1 matrix