Add the casting operation.

2025-06-20 04:00:28 +00:00 · 2023-06-23 21:22:07 +01:00
parent 8ed350dc94
commit 5d44e76e3f
6 changed files with 83 additions and 4 deletions
--- a/src/op.rs
+++ b/src/op.rs
@ -21,6 +21,7 @@ pub(crate) enum Op {
        mul: f64,
        add: f64,
    },
+    ToDType(Tensor),
    Exp(Tensor),
    Log(Tensor),
    Sin(Tensor),