TensorPlay
API symbolsautograd
Copy
View Markdown

tensorplay.autograd.graph.save_on_cpu

tensorplay.autograd.graph.save_on_cpu(pin_memory: bool = False, device_type: str = 'cuda')[source]

Store saved values in host memory and restore their original device.

On this page

Ask DeepWiki