tensor_ops

Functions

fn tensor_add(a: i64, b: i64, out: i64)

fn tensor_sub(a: i64, b: i64, out: i64)

fn tensor_mul_elem(a: i64, b: i64, out: i64)

fn tensor_scale(a: i64, s: i64, out: i64)

fn tensor_sum(a: i64) -> i64

fn tensor_mean(a: i64) -> i64

fn tensor_max(a: i64) -> i64

fn tensor_relu(a: i64, out: i64)

fn tensor_sigmoid(a: i64, out: i64)

fn tensor_tanh_t(a: i64, out: i64)

fn tensor_softmax(a: i64, out: i64)

fn tensor_transpose(a: i64, out: i64)

fn matmul(a: i64, b: i64, out: i64)