Add the cat operator (without the storage implementation for now).

This commit is contained in:
laurent
2023-06-23 10:13:37 +01:00
parent bf9e1d1c23
commit 6110db31c9
4 changed files with 118 additions and 0 deletions

View File

@ -8,6 +8,8 @@ pub(crate) enum Op {
Div(Tensor, Tensor),
Matmul(Tensor, Tensor),
Cat(Vec<Tensor>, usize),
#[allow(dead_code)] // add is currently unused.
Affine {
arg: Tensor,