Nx

Mix.install([
  {:nx, "~> 0.6.4"}
])

Tensor Creation

data = [[1, 2], [3, 4], [5, 6]]
tensor = data |> Nx.tensor()

Tensor Operations

tensor |> Nx.to_heatmap()

tensor |> Nx.shape()

tensor |> Nx.exp()

# softmax
Nx.divide(Nx.exp(tensor), Nx.sum(Nx.exp(tensor)))