MCPcopy Create free account
hub / github.com/branislav1991/PyTorchProjectFramework / transfer_to_device

Function transfer_to_device

utils/__init__.py:10–19  ·  view source on GitHub ↗

Transfers pytorch tensors or lists of tensors to GPU. This function is recursive to be able to deal with lists of lists.

(x, device)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

set_inputMethod · 0.90

Calls

no outgoing calls

Tested by

no test coverage detected