cerebras.modelzoo.common.pytorch_utils.to_cpu#

cerebras.modelzoo.common.pytorch_utils.to_cpu(tensor)[source]#

Move tensor from device to cpu.

previous

cerebras.modelzoo.common.pytorch_utils

next

cerebras.modelzoo.common.pytorch_utils.to_tensor