我正在尝试定义一个返回 2 x 2 矩阵的函数。具体来说,我有:
def f(d,n):
return scipy.mat([[1,d/n],[0,1]])
当 d 和 n 是标量输入时,这可以正常工作。但如果 d 和 n 是向量输入,则此功能不起作用。我不知道如何向量化这个函数;例如,我想做一个函数 f 执行以下操作:
认为
d=[1,2,3,4]
n=[5,6,7,8],
f(d,n)
返回一个由 4 个矩阵组成的数组:
[[[1,1/5],[0,1]],[[1,2/6],[0,1]],[[1,3/7],[0,1]],[[1,4/8],[0,1]]]
如果你能告诉我这是否可能,那就太棒了!谢谢!