cupy.memoize

cupy.memoize(bool for_each_device=False)

使一个函数记忆每个参数和设备的结果。

这个装饰器提供了函数结果的自动记忆化。

参数:

for_each_device (bool) – 如果 True，它会在每个设备上记忆结果。否则，它只会根据参数记忆结果。