Cleanup the tensor creation code.

2025-06-20 04:00:28 +00:00 · 2023-06-23 19:52:21 +01:00
parent 88187b784b
commit fe75a01188
2 changed files with 37 additions and 116 deletions
--- a/src/op.rs
+++ b/src/op.rs
@ -23,6 +23,8 @@ pub(crate) enum Op {
    },
    Neg(Tensor),
    Reshape(Tensor),
+    #[allow(dead_code)]
+    Softmax(Tensor, usize),
    Sqr(Tensor),
    Sqrt(Tensor),
    ToDevice(Tensor),