pytorch官方文档60分钟入门笔记

文章目录

  • 1. 张量(Tensors)
    • 定义张量
    • 张量操作
  • 2.自动求导(autograd)
    • 变量Variable
  • 3.神经网络
  • 4. 训练一个分类器
    • 载入数据
  • 5.数据并行

day63
参考:官方文档
https://blog.csdn.net/u014630987/article/details/78669051

1. 张量(Tensors)

tensors和numpy的ndarray类似,但是tensors可以使用GPU加快运算

张量定义的matrix在未使用前的初始值就是那个内存地址保存的值

定义张量

#定义未初始化的张量
x = torch.empty(5,3);
#定义初始化的张量
x = torch.rand(5,3);#随机初始化
x = torch.zeros(5,3,dtype=torch.long);#全零初始化
x = torch.tensor([5.5,3])#直接从数值构建tensor
#从已有的张量构建新的张量,会复用已有张量的property,除非指定新的
x = x.new_ones(5,3,dtype=torch.double)
#输出张量矩阵的大小
print(x.size())#返回torch.Size([5, 3]),是tuple类型

张量操作

相加:

#语法一
y = torch.rand(5,3)
print(x+y)
#语法二
print(torch.add(x,y));
#语法三:指定out张量
result = torch.empty

你可能感兴趣的:(机器学习)