torch.cuda.memory.caching\_allocator\_alloc =========================================== .. currentmodule:: torch.cuda.memory .. autofunction:: caching_allocator_alloc