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.setup_logging

next

cerebras.modelzoo.common.pytorch_utils.to_tensor