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 -> TensorLayout
Parameters
Type | Name | Description |
---|---|---|
int | ax1 | |
int | ax2 | |
TensorLayout | a |
Returns
Type | Description |
---|---|
TensorLayout |