PyTorch系列(一) 60mins - Autograd
目录AutogradPyTorch: Tensors and autogradPyTorch: Defining new autograd functions
Autograd
PyTorch: Tensors and autograd
# -*- coding: utf-8 -*-
import torch
dtype = torch.float
device = torch.device("cpu")
# device = torch.device("cuda:0") # Uncomment this
原创
2020-05-20 14:54:22 ·
315 阅读 ·
0 评论