zyx-derive

本crate包含zyx的过程宏。

宏模块自动为您模块实现IntoIterator，以便您可以在反向传播中使用它并将其保存到磁盘。

use zyx_core::backend::Backend;
use zyx_core::tensor::Tensor;

#[derive(Module)]
struct MyNet<B: Backend> {
    b: Tensor<B>,
    w: Tensor<B>,
}

impl<B: Backend> MyNet<B> {
    fn forward(&self, x: &Tensor<B>) -> Tensor<B> {
        x.dot(self.w) + self.b
    }
}

有关README和源代码，请访问 github。

有关更多详细信息，请参阅 书籍。

特性

• std - 启用 zyx-core/std

lib.rs:

zyx-derive

本crate包含zyx的过程宏。

宏模块自动为您模块实现IntoIterator，以便您可以在反向传播中使用它并将其保存到磁盘。

use zyx_core::backend::Backend;
use zyx_core::tensor::Tensor;
use zyx_derive::Module;

#[derive(Module)]
struct MyNet<B: Backend> {
    b: Tensor<B>,
    w: Tensor<B>,
}

impl<B: Backend> MyNet<B> {
    fn forward(&self, x: &Tensor<B>) -> Tensor<B> {
        x.dot(&self.w) + &self.b
    }
}

有关README、快速教程和源代码，请访问 https://www.github.com/zk4x/zyx。

有关更多详细信息，请参阅 书籍。