当前位置:   article > 正文

每天五分钟玩转深度学习PyTorch:创建pytorch中的零维标量tensor

每天五分钟玩转深度学习PyTorch:创建pytorch中的零维标量tensor

标量是什么?

tensor张量是一个多维数组,零维就是一个点(就是本章的标量),一维就是向量,二维就是一般的矩阵,多维就相当于一个多维的数组,这和 numpy理解是一样的,不同的是Tensor不仅可以在CPU上跑,在GPU上也可以跑。

标量(scalar),只具有数值大小,而没有方向,我们可以把标量看成是一个实数。本节课程我们学习如何使用pytorch来创建标量,在pytorch中只要dim=0的tensor,我们就可以认为它是标量。

创建标量

  1. import torch
  2. import numpy as np
  3. a=torch.tensor(1.)#创建FLoatTensor类型的标量
  4. print(a)
  5. print(a.dim())
  6. print(a.type())
  7. print(a.shape)
  8. print(a.size())
  9. print("--------------------------------")
  10. a=torch.tensor(1)#创建LongTensor类型的标量
  11. print(a)
  12. print(a.dim())
  13. print(a.type())
  14. print(a.shape)
  15. print(a.size())
  16. print("--------------------------------")
  17. a=torch.tensor([1])#创建LongTensor类型的向量
  18. print(a)
  19. print(a.dim())
  20. print(a.type())
  21. print(a.shape)
  22. print(a.size())
  23. p
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/很楠不爱3/article/detail/513796
推荐阅读
相关标签
  

闽ICP备14008679号