XArray apply_ufunc用于具有多个输出的函数？

Question

一个简单的问题，但是无法绕过文档：

给定两个DataArray，如何使用apply_ufunc以便以新的维度收集函数的输出？

例如：

test1 = xr.DataArray(np.linspace(1, 6, 6).reshape(3, 2))
test2 = xr.DataArray(np.linspace(6, 1, 6).reshape(3, 2))

def foo(a, b):
    return a+b, a-b

xr.apply_ufunc(sum, test1, test2)

这将返回错误： ValueError：应用的函数返回的数据具有意外的维数：维数（'dim_0'，'dim_1'）为3 vs 2

关于如何执行此操作的任何想法？