Deeplearning specialization向量化代码(亲测可用)

这是vectorization课程学习的代码笔记,vscode+python编写,另外小虎也在MATLAB使用bsxfun函数进行广播。

时间对比

import time
import numpy as np

a = np.random.rand(1000000)
b = np.random.rand(1000000)

tic = time.time()
c = np.dot(a, b)
toc = time.time()

print(c)
print("vectorized version:" + str(1000*(toc-tic)) + "ms")

c = 0
tic = time.time()
for i in range(1000000):
    c += a[i]*b[i]
toc = time.time()

print(c)
print("vectorized version:" + str(1000*(toc-tic)) + "ms")

封装函数算法

import time
import numpy as np

import math

v = np.array([1, 2, 3, 4])

tic = time.time()
u1 = np.zeros((4, 1))
for i in range(4):
    u1[i] = math.exp(v[i])
toc = time.time()

print(u1)
print(str(1000 * (toc-tic)) + "ms")

tic = time.time()
u2 = np.exp(v)
toc = time.time()

print(u2)
print(str(1000 * (toc-tic)) + "ms")

# np.log(v), np.abs(v) ,np.maximum(v,0), v**2

brocasting广播

import time
import numpy as np

A = np.array([[56.0, 0.0, 4.4, 68.0],
             [1.2, 104.0, 52.0, 8.0],
             [1.8, 135.0, 99.0, 0.9]])

cal=A.sum(axis=0)
print(cal)

percentage=100*A/cal.reshape(1,4)
# percentage=100*A/cal
print(percentage)

用bsxfun广播

A=[[56.0, 0.0, 4.4, 68.0];
    [1.2, 104.0, 52.0, 8.0];
    [1.8, 135.0, 99.0, 0.9]];

cal=sum(A,1)

fun=@(a,b)100*a./b;
percentage=bsxfun(fun,A,cal)

在这里插入图片描述
其实小虎觉得没必要用bsxfun,如下,直接算就好了。

A=[[56.0, 0.0, 4.4, 68.0];
    [1.2, 104.0, 52.0, 8.0];
    [1.8, 135.0, 99.0, 0.9]];

cal=sum(A,1)

percentage=100*A./cal
posted @   狂小虎  阅读(29)  评论(0编辑  收藏  举报
编辑推荐:
· 基于Microsoft.Extensions.AI核心库实现RAG应用
· Linux系列:如何用heaptrack跟踪.NET程序的非托管内存泄露
· 开发者必知的日志记录最佳实践
· SQL Server 2025 AI相关能力初探
· Linux系列:如何用 C#调用 C方法造成内存泄露
阅读排行:
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 别再用vector<bool>了!Google高级工程师:这可能是STL最大的设计失误
· 单元测试从入门到精通
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 上周热点回顾(3.3-3.9)
点击右上角即可分享
微信分享提示