NX Examples

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

Basic Tensor Creation and Manipulation

t = Nx.tensor([[1, 2], [3, 4]])

Nx.shape(t)

xt = Nx.exp(t)
result = Nx.divide(xt, Nx.sum(xt))