cupy.memoize

cupy.memoize(bool for_each_device=False)

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

此装饰器提供函数结果的自动记忆功能。

参数:

for_each_device (bool) – 如果为 True，它将记忆每个设备的结果。否则，它仅根据参数记忆结果。