nodev 1.0.0__py3-none-any.whl

nodev/utils.py

@@ -0,0 +1,172 @@
+
+import  os
+import subprocess
+def _dot_func(f):
+    txt = ''
+    func_name = f.__class__.__name__
+
+    # 函数节点（蓝色）
+    txt += '{} [label="{}", color=lightblue, style=filled]\n'.format(
+        id(f), func_name
+    )
+
+    # 输入 -> 函数
+    for x in f.inputs:
+        txt += '{} -> {}\n'.format(id(x), id(f))
+
+    # 函数 -> 输出
+    for y in f.outputs:
+        txt += '{} -> {}\n'.format(id(f), id(y()))
+
+    return txt
+
+def _dot_var(v,verbose=False):
+    dot_var = '{} [label="{}", color=orange, style=filled]\n'
+    name=''if v.name is None else v.name
+    if verbose and v.data is not None:
+        if v.name is not None:
+            name+=':'
+        name+=str(v.shape)+' '+str(v.dtype)
+    return dot_var.format(id(v),name)
+def max_backward_shape(x_shape, axis):
+    if axis is None:
+        return (1,) * len(x_shape)
+
+    if not isinstance(axis, tuple):
+        axis = (axis,)
+
+    shape = list(x_shape)
+
+    for a in axis:
+        shape[a] = 1
+
+    return tuple(shape)
+def get_dot_graph(output,verbose=True):
+    txt=''
+    funcs=[]
+    seen_set=set()
+
+    def add_func(f):
+        if f not in seen_set:
+            funcs.append(f)
+            seen_set.add(f)
+    add_func(output.creator)
+    txt+=_dot_var(output,verbose)
+    while funcs:
+        func=funcs.pop()
+        txt+=_dot_func(func)
+
+        for x in func.inputs:
+            txt+=_dot_var(x,verbose)
+
+            if x.creator is not None:
+                add_func(x.creator)
+    print(txt)
+    return 'digraph g {\n' + txt + '}'
+def reshape_sum_backward(gy, x_shape, axis, keepdims):
+    ndim = len(x_shape)
+
+    if axis is None:
+        return gy.reshape((1,) * ndim)
+
+    if not isinstance(axis, tuple):
+        axis = (axis,)
+
+    if not keepdims:
+        actual_axis = [a if a >= 0 else a + ndim for a in axis]
+        shape = list(gy.shape)
+
+        for a in sorted(actual_axis):
+            shape.insert(a, 1)
+
+        gy = gy.reshape(shape)
+
+    return gy
+def plot_dot_graph(output,verbose=True,to_file='graph.png'):
+    dot_graph=get_dot_graph(output,verbose)
+
+    tmp_dir=os.path.join(os.path.expanduser('~'),'.nodev')
+    if not os.path.exists(tmp_dir):
+        os.mkdir(tmp_dir)
+    graph_path=os.path.join(tmp_dir,'tmp_graph.dot')
+
+    with open(graph_path,'w') as f:
+        f.write(dot_graph)
+
+    extension=os.path.splitext(to_file)[1][1:]
+    cmd = 'dot -T{} {} -o {}'.format(extension, graph_path, to_file)
+    subprocess.run(cmd,shell=True)
+    print(cmd)
+
+
+def sum_to(x, shape):
+    """
+    将数组 x 沿着广播过的维度求和，使其形状变为 shape
+    """
+    ndim = len(shape)
+    lead = x.ndim - ndim
+
+    # 需要被压缩（求和）的前导轴
+    lead_axis = tuple(range(lead))
+
+    # 在 shape 中为 1 的轴，也需要被求和
+    axis = tuple(
+        i + lead for i, sx in enumerate(shape) if sx == 1
+    )
+
+    y = x.sum(lead_axis + axis, keepdims=True)
+
+    # 去掉多余的维度
+    if lead > 0:
+        y = y.squeeze(lead_axis)
+
+    return y  #补充函数，正文没写
+
+
+import os
+import urllib.request
+
+cache_dir = os.path.join(os.path.expanduser('~'), '.dezero')
+
+def pair(x):
+    if isinstance(x,int):
+        return (x,x)
+    elif isinstance(x,tuple):
+        assert len(x)==2
+        return x
+    else:
+        raise ValueError
+def get_file(url, file_name=None):
+    """Download a file if it is not in the cache."""
+    cache_dir = "../data"
+    if file_name is None:
+        file_name = url[url.rfind('/') + 1:]
+
+    file_path = os.path.join(cache_dir, file_name)
+
+    # 如果缓存目录不存在就创建
+    if not os.path.exists(cache_dir):
+        os.mkdir(cache_dir)
+
+    # 如果文件已经存在直接返回
+    if os.path.exists(file_path):
+        return file_path
+
+    print("加载: " + file_name)
+
+    try:
+        urllib.request.urlretrieve(url, file_path)
+    except (Exception, KeyboardInterrupt):
+        # 如果下载失败删除不完整文件
+        if os.path.exists(file_path):
+            os.remove(file_path)
+        raise
+
+    print("Done")
+
+    return file_path
+def get_conv_outsize(size, k, s, p):
+    return (size + 2 * p - k) // s + 1
+
+
+#print(pair(1),pair((1,4)))

nodev-1.0.0.dist-info/METADATA

@@ -0,0 +1,14 @@
+Metadata-Version: 2.1
+Name: nodev
+Version: 1.0.0
+Summary: 一个轻量级深度学习第三方库，2026年设计。
+Author-email: Yang Yaqing <youngtoothh@foxmail.com>
+License: MIT
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: requests >=2.28.0
+Requires-Dist: numpy
+

nodev-1.0.0.dist-info/RECORD

@@ -0,0 +1,17 @@
+__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nodev/__init__.py,sha256=_6TL6fjt94ICY750euKt7Fy6dGC5vHy5mQoR1v4arEw,597
+nodev/core.py,sha256=BZN6ewqtGj49cKrFLC_PYGAtfuQhzQiXFH6nkTXn4Xc,9215
+nodev/cuda.py,sha256=wone0xXTZdznQ0f9cZ5iW9JPgMrAk6-ePsVaPvDOoks,630
+nodev/dataloaders.py,sha256=mxs4FGJvBmrFAZ1TVsCxaLUxbmUlTaNwhn-BxRwJEPk,2000
+nodev/dataset.py,sha256=AOUMciTaCDMxpJq-wTCN_D_pLu5uo6uvRDPiP0hgs80,5705
+nodev/layers.py,sha256=KLbP9zkyzFQy-QMD5ntWSq0cuIgrcNFRRhQF8Bh-ZLg,5879
+nodev/model.py,sha256=Y5mVSD1L3367vKMUgcfUQ2lN8qy7BXv2B5BilBSLZS4,4312
+nodev/operations.py,sha256=0-wtkfeyGWCdfXl9rNFMF21aK8q_Dr6JEEbIwS8o3pk,10293
+nodev/operations_conv.py,sha256=PwzrfX5fyHwwJLKKCmxJd9_3rWMjuU_2SXqyeNoU_e4,15855
+nodev/optimizer.py,sha256=nLeZg0STkSl0q4N9mCAE5UJGyhJNVCwdIaPxyZoMvSw,1874
+nodev/transforms.py,sha256=66Y1CslwoIfMVPuWt1fh1oDl4_lJEXAaWwiR1dPllJk,1305
+nodev/utils.py,sha256=SrPbOdSCORFAEUY0_GVro_4MYW24uzHNO3H0g7wDe_o,4321
+nodev-1.0.0.dist-info/METADATA,sha256=Isew386qYzm2p4aWrz77sSJ2K9GZrVp7AS3xxAMa5CM,460
+nodev-1.0.0.dist-info/WHEEL,sha256=BNRMDyzLkkcmlv0J8ppDQkk2VED33SesJDynr9ED1gc,91
+nodev-1.0.0.dist-info/top_level.txt,sha256=mV5FbarWYa_abJuO6m6ZhbgC3rw8Lig5WBvL3xtHa_A,15
+nodev-1.0.0.dist-info/RECORD,,

nodev-1.0.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (75.3.4)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

nodev-1.0.0.dist-info/top_level.txt

@@ -0,0 +1,2 @@
+__init__
+nodev