ローゼンブロック関数に対する勾配法のRust実装です。 元ネタはゼロから作るDeep Learning⑤から。 github.com ソースはこれ。 use burn::tensor::Tensor; use burn::backend::{Autodiff, Wgpu}; type Backend = Wgpu; type AutoDIffBackend = Autodiff<Backend>; fn rosenbrok(x0: &Tensor<AutoDIffBackend, 1>, x1: &Tensor<AutoDIffBackend, 1>) -> Tensor<AutoDIffBa…