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.