Show HN: Glowstick – type level tensor shapes in stable rust
a year ago
- #Tensor
- #Rust
- #Machine Learning
- 该crate通过在类型系统中追踪张量形状,使Rust中的张量操作变得安全、简单且有趣。
- 提供了使用candle的示例,包括matmul(矩阵乘法)、reshape(重塑)、unsqueeze(增加维度)、squeeze(压缩维度)、narrow(窄化)、broadcast_add(广播加法)、transpose(转置)、conv2d(二维卷积)和flatten(展平)等操作。
- 该项目目前处于1.0之前的预发布阶段,意味着将会进行破坏性变更。
- 特性包括:将张量形状表示为类型、支持动态维度、人类可读的错误信息,以及手动检查类型级形状。
- 计划支持所有ONNX操作。