Is the output of the network limited to 1 dimension?

[xvxvxvaduo]

Is the output of the network limited to 1 dimension? If I want the output of the network to be d-dimensional, should I modify the shape of output of the branch net or trunk net to (d, p). The output of branch net and trunk net seems to be a vector, and their dot product is a scalar. If I want the final result to be a vector, it seems that one of the networks needs to output a matrix

[lululxvi]

Check https://doi.org/10.1016/j.cma.2022.114778 and DeepXDE code.