文件名称:tensorcheck:用于机器学习系统的张量的运行时验证
文件大小:6KB
文件格式:ZIP
更新时间:2024-05-20 20:10:15
Python
TensorCheck pip install tensorcheck 用于机器学习系统的张量的运行时验证。 这是通过将函数装饰器用作自省的钩子来验证函数的张量输入的幼稚方法。 这并非设计为可组合的类型系统,而是仅允许简洁地描述设计新架构或编写可视化代码时预期的张量。 支持: 用于np.ndarray和torch.Tensor Tensor dtypes验证 形状验证,包括通用形状变量 输入验证范围 用法示例 import numpy as np import torch from tensorcheck import tensorcheck @ tensorcheck ({ "img" : { "dtype" : np . uint8 , "shape" : [ 1 , 3 , "H" , "W" ], "range" :
【文件预览】:
tensorcheck-main
----pyproject.toml(90B)
----MIT_LICENSE.txt(1KB)
----example.py(1KB)
----setup.cfg(483B)
----.gitignore(226B)
----README.md(2KB)
----tensorcheck()
--------test.py(8KB)
--------tensorcheck.py(4KB)
--------__init__.py(27B)