Current section
Files
Jump to
Current section
Files
lib/bumblebee/utils/model.ex
defmodule Bumblebee.Utils.Model do
@moduledoc false
@doc """
Adds another word to a hierarchical name.
"""
@spec join(String.t() | nil, String.Chars.t()) :: String.t()
def join(name, suffix)
def join(nil, suffix), do: to_string(suffix)
def join(name, suffix), do: name <> "." <> to_string(suffix)
@doc """
Converts a list of inputs to a map with input names as keys.
"""
@spec inputs_to_map(list(Axon.t())) :: %{String.t() => Axon.t()}
def inputs_to_map(inputs) when is_list(inputs) do
for %Axon{output: id, nodes: nodes} = axon <- inputs, into: %{} do
input = nodes[id]
{input.name.(:input, %{}), axon}
end
end
end