nx

Types

pub type NxTensor =
  Dynamic

Functions

pub fn add(a: Dynamic, b: Dynamic) -> Dynamic
pub fn broadcast(a: Float, shape: a) -> Dynamic
pub fn divide(a: Dynamic, b: Float) -> Dynamic
pub fn dot(a: Dynamic, b: Dynamic) -> Dynamic
pub fn eye(n: Int) -> Dynamic
pub fn indexed_put(
  a: Dynamic,
  indices: Dynamic,
  value: Float,
) -> Dynamic
pub fn map(a: Dynamic, f: fn(Dynamic) -> Float) -> Dynamic
pub fn multiply(a: Dynamic, b: Float) -> Dynamic
pub fn new_axis(a: Dynamic, axis: Int) -> Dynamic
pub fn norm(a: Dynamic) -> Dynamic
pub fn put_slice(
  a: Dynamic,
  indices: List(Int),
  value: Dynamic,
) -> Dynamic
pub fn shape(a: Dynamic) -> #(Int)
pub fn solve(a: Dynamic, b: Dynamic) -> Dynamic
pub fn subtract(a: Dynamic, b: Dynamic) -> Dynamic
pub fn tensor(a: List(a)) -> Dynamic
pub fn to_list_1d(a: Dynamic) -> List(Float)
pub fn to_list_2d(a: Dynamic) -> List(List(Float))
pub fn to_number(a: Dynamic) -> Float
pub fn transpose(a: Dynamic) -> Dynamic
Search Document