Function mit array als input und outout

Hallo zusammen, ich verzweifel gerade hier dran: ich moechte eine funktion die ein Array bildet, dessen Diagonale aus Werten berechnet wird, die ich der Funktion vorgebe (auch als Array). der Rest des (ausgabe-)Arrays ist null.

Bisher habe ich das geschrieben: Public Function createSplineArray(Steps() As Double) As Double
Dim i As Integer
Dim ArrayOne(UBound(Steps(1)), UBound(Steps(1))) as Double
For i = 1 To UBound(Steps(1))
ArrayOne(i, i) = 2 * (Steps(1) + Steps(2))
Next i
createSplineArray() = ArrayOne()
End Function

aus dem Input  1   soll werden: 4  0  0
                         1                        0  4  0
                         1                        0  0  4

Danke fuer eure hilfe!