Skip to content

Instantly share code, notes, and snippets.

@shnhrtkyk

shnhrtkyk/get_pytorch_tensor.py

Created June 4, 2019 22:01
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save shnhrtkyk/29dfa7a3c30567e1969199610d065d49 to your computer and use it in GitHub Desktop.
Save shnhrtkyk/29dfa7a3c30567e1969199610d065d49 to your computer and use it in GitHub Desktop.
Download ZIP
PytorchのCUDA tensorをnumpyに変換
Raw
get_pytorch_tensor.py
cudatensor.detach().cpu().numpy()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment