Cannot cast ufunc subtract output from・・・というエラーが出て困っています

Question

前提・実現したいこと

pythonで勾配法を試しています。

下のコードの「gradient_descent(function_2,init_x=init_x,lr=0.1,step_num=100)」
という部分を入力した際、以下のエラーメッセージが発生しました。

発生している問題・エラーメッセージ

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 7, in gradient_descent
TypeError: Cannot cast ufunc subtract output from dtype('float64') to dtype('int64') with casting rule 'same_kind'

該当のソースコード

def function_2(x):
	return x[0]**2 + x[1]**2
def numerical_gradient(f,x): 
	h = 1e-4
	grad = np.zeros_like(x)
	
	for idx in range(x.size):
		tmp_val = x[idx]
		x[idx] = tmp_val + h
		fxh1 = f(x)
		
		x[idx] = tmp_val - h
		fxh2 = f(x)
		
		grad[idx] = (fxh1 - fxh2) / (2*h)
		x[idx] = tmp_val
		
	return grad

	
def gradient_descent(f, init_x, lr=0.01, step_num=100):
	x = init_x
	
	for i in range(step_num):
		grad = numerical_gradient(f, x)
		x -= lr * grad
	return x

init_x = np.array([-3,4])

gradient_descent(function_2,init_x=init_x,lr=0.1,step_num=100)

試したこと

「Cannot cast ufunc ・・・」で検索してみましたが日本語のページがなく情報を得られませんでした
どなたかご教示いただけませんでしょうか

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Answer 1

変数 x に格納されているデータ型が整数型だからでしょう。
同様のエラーはこんなふうに再現できます。

>>> arr = np.array([1, 2])
>>> arr -= np.array([1.0, 2.0])
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: Cannot cast ufunc subtract output from dtype('float64') to dtype('int32') with casting rule 'same_kind'

解決するためには、xに格納するデータ型を明示的に決めれば良いです。

init_x = np.array([-3, 4], dtype=np.float)

Comments

[50002]

うまくいきました！
intかfloatかを明示することが大事なんですね、
勉強になりました、わかりやすいご回答ありがとうございます！