使用 cupy 时变量值无故改变

Question

我正在使用 cupy 实现机器学习的梯度公式。但是我遇到了一个奇怪的错误，变量 temp_value 无缘无故地改变了。这是我的代码：

import cupy as cp
def gradient_1d(f, x):
    grad = cp.zeros_like(x)
    h = 0.00001
    for index in range(x.size):
        temp_value = x[index]
        print(temp_value,'1')
        x[index] = float(temp_value) + h
        print(temp_value,'2')
        plus_result = f(x)
        print(temp_value,'3')
        x[index] = temp_value - h
        print(temp_value,'4')
        minus_result = f(x)
        print(temp_value,'5')
        grad[index] = ((plus_result - minus_result)/2*h)
        print(temp_value,'6')
        x[index] = temp_value
    return grad


def f2(x):
    if x.ndim == 1:
        return cp.sum(x**2)
    else:
        return cp.sum(x**2, axis=1)


def main():
    x = cp.array([1, 2])
    grad = gradient_2d(f2, x)
    print(grad)


main()

输出是：

您可以观察到 temp_value 的值在每次到达第 4 个检查点时减一，但我没有做任何更新此变量的操作。

为什么？？？提前感谢您的帮助！

Answer 1

这可能是因为 temp 被分配给 x[index] 但 x[index] 被更改了。使用 temp_value = x[index].copy() 而不是 temp_value = x[index] 作为 for 循环中的第一个语句。为此，您还需要在文件顶部添加 import copy。

使用 cupy 时变量值无故改变

Variable value change for no reason when using cupy

python

gradient

machine-learning

cupy