Packages

gleastsq

0.1.0
0.6.3 0.6.2 0.5.2 0.5.1 0.5.0 0.4.3 0.4.2 0.4.1 0.4.0 0.3.2 0.3.1 0.3.0 0.2.0 0.1.2 0.1.1 0.1.0

A least squares curve fitting library for Gleam.

HexDocs

Current section

Files

Jump to
Readme
16 Versions
2 Dependencies
Dependants
Files
Activity
gleastsq src nx.erl
Raw

src/nx.erl 

-module(nx).

-compile([no_auto_import, nowarn_unused_vars, nowarn_unused_function, nowarn_nomatch]).



-export([tensor/1, eye/1, dot/2, map/2, multiply/2, add/2, subtract/2, transpose/1, shape/1, to_list_1d/1, to_list_2d/1, to_number/1, norm/1, solve/2, broadcast/2, indexed_put/3, new_axis/2, divide/2, put_slice/3]).



-spec tensor(list(any())) -> gleam@dynamic:dynamic_().

tensor(A) ->

    'Elixir.Nx':tensor(A).



-spec eye(integer()) -> gleam@dynamic:dynamic_().

eye(N) ->

    'Elixir.Nx':eye(N).



-spec dot(gleam@dynamic:dynamic_(), gleam@dynamic:dynamic_()) -> gleam@dynamic:dynamic_().

dot(A, B) ->

    'Elixir.Nx':dot(A, B).



-spec map(gleam@dynamic:dynamic_(), fun((gleam@dynamic:dynamic_()) -> float())) -> gleam@dynamic:dynamic_().

map(A, F) ->

    'Elixir.Nx':map(A, F).



-spec multiply(gleam@dynamic:dynamic_(), float()) -> gleam@dynamic:dynamic_().

multiply(A, B) ->

    'Elixir.Nx':multiply(A, B).



-spec add(gleam@dynamic:dynamic_(), gleam@dynamic:dynamic_()) -> gleam@dynamic:dynamic_().

add(A, B) ->

    'Elixir.Nx':add(A, B).



-spec subtract(gleam@dynamic:dynamic_(), gleam@dynamic:dynamic_()) -> gleam@dynamic:dynamic_().

subtract(A, B) ->

    'Elixir.Nx':subtract(A, B).



-spec transpose(gleam@dynamic:dynamic_()) -> gleam@dynamic:dynamic_().

transpose(A) ->

    'Elixir.Nx':transpose(A).



-spec shape(gleam@dynamic:dynamic_()) -> {integer()}.

shape(A) ->

    'Elixir.Nx':shape(A).



-spec to_list_1d(gleam@dynamic:dynamic_()) -> list(float()).

to_list_1d(A) ->

    'Elixir.Nx':to_list(A).



-spec to_list_2d(gleam@dynamic:dynamic_()) -> list(list(float())).

to_list_2d(A) ->

    'Elixir.Nx':to_list(A).



-spec to_number(gleam@dynamic:dynamic_()) -> float().

to_number(A) ->

    'Elixir.Nx':to_number(A).



-spec norm(gleam@dynamic:dynamic_()) -> gleam@dynamic:dynamic_().

norm(A) ->

    'Elixir.Nx.LinAlg':norm(A).



-spec solve(gleam@dynamic:dynamic_(), gleam@dynamic:dynamic_()) -> gleam@dynamic:dynamic_().

solve(A, B) ->

    'Elixir.Nx.LinAlg':solve(A, B).



-spec broadcast(float(), any()) -> gleam@dynamic:dynamic_().

broadcast(A, Shape) ->

    'Elixir.Nx':broadcast(A, Shape).



-spec indexed_put(gleam@dynamic:dynamic_(), gleam@dynamic:dynamic_(), float()) -> gleam@dynamic:dynamic_().

indexed_put(A, Indices, Value) ->

    'Elixir.Nx':indexed_put(A, Indices, Value).



-spec new_axis(gleam@dynamic:dynamic_(), integer()) -> gleam@dynamic:dynamic_().

new_axis(A, Axis) ->

    'Elixir.Nx':new_axis(A, Axis).



-spec divide(gleam@dynamic:dynamic_(), float()) -> gleam@dynamic:dynamic_().

divide(A, B) ->

    'Elixir.Nx':divide(A, B).



-spec put_slice(

    gleam@dynamic:dynamic_(),

    list(integer()),

    gleam@dynamic:dynamic_()

) -> gleam@dynamic:dynamic_().

put_slice(A, Indices, Value) ->

    'Elixir.Nx':put_slice(A, Indices, Value).