问题描述
神奇,这里 2 - 0.0001 居然得到的结果是 1 ?而且我不明确这种问题会在什么情况下发生,我感到很害怕。x_grad = numerical_gradient_no_batch(function_1, x)
这里如果我传入的x是小数,则是程序是符合预期的。若 x 为整数,就会发生上述问题。
x为整数的运行结果
x为小数的运行结果
环境
python3.8,jupyter notebook,win10操作系统
相关代码
import numpy as np
import matplotlib.pylab as plt
from mpl_toolkits.mplot3d import Axes3D
def function_1(x):
if x.ndim == 1:
return np.sum(x**2)
return np.sum(x**2, axis=1)
# 数值法求函数 f 在某点 x(一个向量) 处的梯度
def numerical_gradient_no_batch(f, x):
h = 1e-4
grad = np.zeros_like(x)
for idx in range(x.size):
tmp_x = x[idx]
x[idx] = tmp_x + h
fh1 = f(x)
print(tmp_x, h, tmp_x + h, x[idx], fh1)
x[idx] = tmp_x - h
fh2 = f(x)
print(tmp_x, h, tmp_x - h, x[idx], fh2)
grad[idx] = (fh1 - fh2) / (h * 2)
x[idx] = tmp_x
print(grad)
return grad
x = np.array([2])
x_grad = numerical_gradient_no_batch(function_1, x)