PYTORCH-what is pytorch 笔记

原文:http://pytorch.org/tutorials/beginner/blitz/tensor_tutorial.html

  • 声明一个张量
  •   x = torch.Tensor(5,3)

  生成一个5*3的为初始化的张量

  •   x = torch.rand(5,3)

  生成一个5*3的全部元素随机的张量

  •   print(x.size())

  打印x的大小,那肯定是5*3,但是他的显示形式如下,而且这也是一个元祖,那么也就是说是一个1*2的张量。原文:torch.Size is in fact a   uple, so it supports the same operations。

  Torch.Size([5,3])

  • 实现一个两个张量的加法,有四种实现方法:
  1. print(x + y)
  2. print(torch.add(x + y))
  3. result = torch.Tensor(5, 3)  
    torch.add(x, y, out = result)
    print(result)
  4. y.add_(x)
    print(y)

       那么可以看到1,2,3都没有改变x或者y的值,那么4通过下划线改变了y的值。

  • 如果用numpy那么很重要的一点就是,numpy和pytorch的张量共用一个内存。用pytorch改了张量,numpy的array也会跟着改变。

 

posted @ 2018-01-02 21:00  甜豆儿  阅读(163)  评论(0编辑  收藏  举报