Functions
Details
tensor_reshape_1d
fn tensor_reshape_1d(t: i64, d0: i64) -> i64
tensor_reshape_2d
fn tensor_reshape_2d(t: i64, d0: i64, d1: i64) -> i64
tensor_slice1d
fn tensor_slice1d(t: i64, start: i64, stop: i64, step: i64) -> i64
tensor_slice2d
fn tensor_slice2d(t: i64, row_start: i64, row_stop: i64, col_start: i64, col_stop: i64) -> i64
tensor_row
fn tensor_row(t: i64, row: i64) -> i64
tensor_col
fn tensor_col(t: i64, col: i64) -> i64
broadcast_add_1d
fn broadcast_add_1d(a: i64, b: i64, out: i64)
broadcast_add_2d1d
fn broadcast_add_2d1d(a: i64, b: i64, out: i64)
broadcast_add_2d_col
fn broadcast_add_2d_col(a: i64, b: i64, out: i64)
broadcast_mul_1d
fn broadcast_mul_1d(a: i64, b: i64, out: i64)
broadcast_mul_2d1d
fn broadcast_mul_2d1d(a: i64, b: i64, out: i64)
broadcast_mul_2d_col
fn broadcast_mul_2d_col(a: i64, b: i64, out: i64)
broadcast_scalar_add
fn broadcast_scalar_add(t: i64, s: i64, out: i64)
broadcast_scalar_mul
fn broadcast_scalar_mul(t: i64, s: i64, out: i64)
broadcast_scalar_sub
fn broadcast_scalar_sub(t: i64, s: i64, out: i64)
broadcast_scalar_div
fn broadcast_scalar_div(t: i64, s: i64, out: i64)
tensor_sum_axis0
fn tensor_sum_axis0(t: i64, rows: i64, cols: i64) -> i64
tensor_sum_axis1
fn tensor_sum_axis1(t: i64, rows: i64, cols: i64) -> i64
tensor_mean_axis0
fn tensor_mean_axis0(t: i64, rows: i64, cols: i64) -> i64
tensor_mean_axis1
fn tensor_mean_axis1(t: i64, rows: i64, cols: i64) -> i64
tensor_max_axis0
fn tensor_max_axis0(t: i64, rows: i64, cols: i64) -> i64
tensor_max_axis1
fn tensor_max_axis1(t: i64, rows: i64, cols: i64) -> i64
tensor_concat1d
fn tensor_concat1d(a: i64, b: i64) -> i64
tensor_vstack
fn tensor_vstack(a: i64, b: i64, cols: i64) -> i64
tensor_hstack
fn tensor_hstack(a: i64, b: i64) -> i64
tensor_where
fn tensor_where(cond: i64, a: i64, b: i64, out: i64)
tensor_clamp
fn tensor_clamp(t: i64, lo: i64, hi: i64, out: i64)
tensor_arange
fn tensor_arange(start: i64, stop: i64, step: i64) -> i64
tensor_linspace
fn tensor_linspace(start: i64, stop: i64, n: i64) -> i64
tensor_copy_1d
fn tensor_copy_1d(t: i64) -> i64
tensor_copy_2d
fn tensor_copy_2d(t: i64) -> i64
