Packages

High-performance numerical computing with Eigen backend

Current section

Files

Jump to
nx_eigen lib nx_eigen.ex
Raw

lib/nx_eigen.ex

defmodule NxEigen do
@moduledoc """
Documentation for `NxEigen`.
"""
@doc """
Returns a new tensor with the given data using the NxEigen backend.
"""
def tensor(data, opts \\ []) do
Nx.tensor(data, Keyword.put(opts, :backend, NxEigen.Backend))
end
end