matonix / dezero Goto Github PK

使用箇所：

• main の test 関数
• 各Functionのpub fn backward内→そもそもこれらは不要？

as is

type Data = Array0<f64>;

• 諸実装がArrayの関数に依存

to be

• type Data を廃止し、多相型による実装に変更
• ndarray に由来する関数群に直接依存しない
• dezero がサポートするプリミティブな操作をトレイト（Tensor）として提供する
• ndarray の配列 struct に対し Tensor の impl をする
  • あるいは https://docs.rs/ndarray/0.15.3/ndarray/trait.NdFloat.html

as is

fn forward_body(x: Vec<Data>) -> Vec<Data>;
fn backward_body(x: Vec<Data>, gy: Vec<Data>) -> Vec<Data>;

のように常に任意の長さを想定した実装

to be

fn forward_body(x: Data, y: Data) -> Data;
fn backward_body(x: Data, gy: Data) -> [Data; 2];

のように実装対象によって柔軟にデータの型を変化できる実装。