ZeNu 层

ZeNu 层是使用Rust实现的神经网络层的集合。它提供了构建神经网络的基本组件，并与ZeNu深度学习库集成。

特性

• 各种层类型，包括全连接（线性）层
• 层参数初始化
• 前向传递计算
• 与ZeNu Autograd的自动微分集成

入门

要在Rust项目中使用ZeNu层，请将以下内容添加到您的Cargo.toml文件中

[dependencies]
zenu-layer = "0.1.0"

以下是使用ZeNu层中的线性层的一个简单示例

use zenu_autograd::creator::from_vec::from_vec;
use zenu_layer::layers::linear::Linear;
use zenu_layer::Layer;

fn main() {
    // Create a new linear layer with input dimension 3 and output dimension 2
    let mut linear_layer = Linear::new(3, 2);

    // Initialize the layer parameters with a random seed
    linear_layer.init_parameters(Some(42));

    // Create input data as a Variable
    let input = from_vec(vec![1., 2., 3.], [1, 3]);

    // Perform a forward pass through the layer
    let output = linear_layer.call(input);

    // Access the layer parameters
    let parameters = linear_layer.parameters();
}

有关更多详细信息和示例，请参阅文档。

许可证

ZeNu层根据MIT许可证授权。