rotta_rs 0.1.0

a Deep Learning library with rust language
Documentation 
1
2
3
4
5
6

use crate::Tensor;

pub fn argmax(tensor: &Tensor, dim: i32) -> Tensor {
    Tensor::from_arrayy(tensor.value.read().unwrap().argmax(dim))
}
// argmax has no derivative