在函数修改后,我想用同一行替换NumPy数组的一行。
这是我的代码:
def _softmax(z):
array = np.exp(z)
array = np.divide(array,np.sum(array))
return array
a = np.array([[1,2,3,4],[5,15,4,7]])
n =_softmax(a[0])
print(n)
a[0]= n
print(a[0])
我得到以下结果:
[0.0320586 0.08714432 0.23688282 0.64391426]
[0 0 0 0]
如您所见,n
可以,但是a[0]
除了[0,0]
之外不会改变。
但是,如果我尝试:
a[0] = np.array([4,1])
...效果很好。