Method diagAxis
val diagAxis: int -> int -> TensorLayout -> TensorLayout
Creates a layout that extracts the diagonal along the given axes. The first axis is replaced with the diagonal and the second axis is removed.
Declaration
val diagAxis: ax1:int -> ax2:int -> a:TensorLayout -> TensorLayoutParameters
| Type | Name | Description |
|---|---|---|
| int | ax1 | |
| int | ax2 | |
| TensorLayout | a |
Returns
| Type | Description |
|---|---|
| TensorLayout |