PyTorch的Tensors和Autograd

来源:互联网 发布:软件实施人员基本素质 编辑:程序博客网 时间:2024/05/30 23:46

PyTorch官网提供了比较多快速入门的tutorial,http://pytorch.org/tutorials/。最经典的是Deep Learning with PyTorch: A 60 Minute Blitz

Tensors

Tensor是torch中的基本数据结构,类比成numpy的ndarray,但可以用GPU加速计算,如torch.Tensor类型的变量调用a.cuda()即可。

Tensor和numpy的互相转换:

b = a.numpy()           # Tensor a -> Numpy ba = torch.from_numpy(b) # Numpy b -> Tensor a

Autograd

自动求导是神经网络训练的必经过程,深度学习框架基本都实现这个模块。

0 0