cupy.memoize#

cupy.memoize(bool for_each_device=False)#

Makes a function memoizing the result for each argument and device.

This decorator provides automatic memoization of the function result.

Parameters

for_each_device (bool) – If True, it memoizes the results for each device. Otherwise, it memoizes the results only based on the arguments.