tch-rs

Rust机器学习之tch-rs

Rust机器学习之tch-rstch-rs是PyTorch接口的Rust绑定，可以认为tch-rs是Rust版的PyTorch。
JarodYv·2023-01-30 03:25

使用AMD显卡安装tch-rs(Rust版Pytorch)

自己研究，制作不宜，转发请标注来源使用环境：6700xt,opensuse15.4,rocm5.4接着上一篇的，如果有不懂的先看上一篇第一步：安装libtorch(直接下pytorch官网的，解压)第二步：安装一个pytorchrocm5.1的(如果你的rocm版本小于rocm5.2则不用管)第三步：根据tch-rsgithub上的页面配置exportLIBTORCH=/path/to/libto
浅析威风·2022-12-06 11:57

tch-rs指南 - Tensor的基本操作

文章目录1概述2Tensor的基本操作2.1Tensor的初始化（1）通过数组创建（2）通过默认方法创建（3）通过其他的`tensor`创建（4）通过`opencv::core::Mat`创建2.2Tensor的属性2.3Tensor的运算（1）改变device（2）获取值(indexingandslicing)（3）合并tensors（4）四则运算参考资料1概述在使用rust进行torch模型部
zjuPeco·2022-10-31 18:47
上一页 1 下一页
按字母分类： ABCDEFGHIJKLMNOPQRSTUVWXYZ其他