Traverses two composite types to see if they are compatible.

For non-composite types, the given fun will be called to compare numbers/tensors pairwise.

Counts the number of non-composite types in the composite type.

examples

Examples

iex> Nx.Defn.Composite.count(123)
1
iex> Nx.Defn.Composite.count({1, {2, 3}})
3
iex> Nx.Defn.Composite.count({Complex.new(1), {Nx.tensor(2), 3}})
3

Flattens the given list of composite types.

Elements that are not tensors (i.e. numbers and Complex numbers) are kept as is unless a custom function is given.

examples

Examples

iex> Nx.Defn.Composite.flatten_list([1, {2, 3}])
[1, 2, 3]

iex> Nx.Defn.Composite.flatten_list([1, {2, 3}], [Nx.tensor(4)])
[1, 2, 3, Nx.tensor(4)]

iex> Nx.Defn.Composite.flatten_list([1, {2, 3}], [Nx.tensor(4)], &Nx.tensor/1)
[Nx.tensor(1), Nx.tensor(2), Nx.tensor(3), Nx.tensor(4)]

Reduces the given composite types with acc and fun.

If composite tensor expressions are given, such as a tuple, the composite type is recursively traversed and returned.

If a non-composite tensor expression is given, the function is invoked for it but not for its arguments.

Traverses the given composite types with fun.

If a composite tensor is given, such as a tuple, the composite type is recursively traversed and returned.

Otherwise the function is invoked with the tensor (be it a number, complex, or actual tensor).

Traverses the given composite types with acc and fun.

If a composite tensor is given, such as a tuple, the composite type is recursively traversed and returned.

Otherwise the function is invoked with the tensor (be it a number, complex, or actual tensor).