文件名称：tensor_compute:使用WebGPU探索GPU计算
文件大小：2.1MB
文件格式：ZIP
更新时间：2024-06-19 04:43:45
Rust
Rust 张量计算 这是一个自学WebGPU计算的个人项目，重点关于机器学习应用程序。 目前的特点： 选择要使用的 GPU（如果系统中超过 1 个） 克隆 (批量) Matmul 雷卢 转置 充满 比较 使连续 [-] 切片（需要更多设计/工作） [-] 索引（需要更多设计/工作） 创建视图张量 进行中： Element Wise Tensor - Tensor Ops： 添加 子 穆尔 标量 - 张量操作： 添加 子 穆尔 工作示例： fn main () { println! ( "Running in {:?}" , GpuStore :: get_default (). info ()); let ma = Tensor :: from_data_and_shape ( vec! [ 1 ., 2 ., 3 ., 4 ., 5
【文件预览】：
tensor_compute-master
----compile_shaders.sh(204B)
----.gitignore(52B)
----README.md(1KB)
----Cargo.toml(518B)
----dev_log.md(200B)
----examples()
--------simple_mm()
--------assignment()
----src()
--------gpu_store()
--------autograd()
--------autograd.rs(6KB)
--------tensors()
--------gpu_internals()
--------lib.rs(487B)
----Cargo.lock(46KB)
----glslc(7.93MB)