-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).