python与向量化小tips

python与向量化小tips,第1张

1.python的广播
  • 让所有输入数组都向其中shape最长的数组看齐,shape中不足的部分都通过在前面加1补齐
  • 输出数组的shape是输入数组shape的各个轴上的最大值
  • 如果输入数组的某个轴和输出数组的对应轴的长度相同或者其长度为1时,这个数组能够用来计算,否则出错
  • 当输入数组的某个轴的长度为1时,沿着此轴运算时都用此轴上的第一组值
2.关于shape的小tips

不易出错,要保证计算不出错,需要使矩阵维度能够进行计算

  1. 为避免出错可以用reshape重新设定矩阵维度:
a.reshape((5,2))
  1. 还可以用assert来判断维度是否符合我们需要的数值
assert(a.shape == (5,2))  # 如果维度不是5*2,则程序在此停止
  1. 定义向量的方法
a = np.random.randn(5,2)  # 需要注意两个维度缺一不可,若要是三个维度,则三个维度缺一不可

欢迎分享,转载请注明来源:内存溢出

原文地址:https://54852.com/langs/918892.html

(0)
打赏 微信扫一扫微信扫一扫 支付宝扫一扫支付宝扫一扫
上一篇 2022-05-16
下一篇2022-05-16

发表评论

登录后才能评论

评论列表(0条)

    保存