Pytorch机器学习——2 安装和快速上手(三)

2022-03-31  本文已影响0人  辘轳鹿鹿

outline

  1. PyTorch安装
  2. Jupyter Notebook使用
  3. NumPy基础知识
  4. PyTorch基础知识

3. NumPy基础知识

3.4 索引、切片和迭代

b[:,1] #每行第2个元素
for row in b: #迭代对第一个轴而言
  print(row)

for element in b.flat: #对数组中的每个元素进行运算
  print(element)
image.png

3.5 数组赋值

会索引就会赋值

3.6 更改数组的形状

a.reshape(3,4) #新建一个符合目标形状的数组
a.ravel() #返回扁平化数组

3.7 组合、拆分数组

np.vstack((a,b))
np.hstack((a,b))
np.column_stack((a,b))
np.hsplit(a,3) #允许指定返回相同形状数组的个数
np.vsplit(x,np.array([3,6])) #指定在哪些列后发生分割
image.png

3.8 广播

广播的原则能使通用函数有意义地处理不具有相同形状的输入

有用就留个赞吧^_^

上一篇 下一篇

猜你喜欢

热点阅读