# View Source Nx.Defn.Composite (Nx v0.6.4)

Functions to deal with composite data types.

Composite data-types are traversed according to `Nx.Container`

.
If a regular tensor is given, it is individually traversed.
Numerical values, such as integers, floats, and complex numbers
are not normalized before hand. Use `Nx.to_tensor/1`

to do so.

The functions in this module can be used both inside and outside `defn`

.
Note that, when a value is given to `defn`

, it is first converted to
tensors and containers via `Nx.LazyContainer`

. Inside `defn`

, there are
no lazy containers, only containers.

# Summary

## Functions

Traverses two composite types to see if they are compatible.

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

Flattens recursively the given list of composite types.

Reduces recursively the given composite types with `acc`

and `fun`

.

Traverses recursively the given composite types with `fun`

.

Traverses recursively the given composite types with `acc`

and `fun`

.

# Functions

Traverses two composite types to see if they are compatible.

Non-tensor values are first compared using `Nx.LazyContainer`

and then, if not available, as `Nx.Container`

.

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

```
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 recursively 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

```
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)]
```

Reduces recursively 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 recursively 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 recursively 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).