Method allIdxOfDim
val allIdxOfDim: int -> TensorLayout -> seq<int64>
all indices of the given dimension
Declaration
val allIdxOfDim: dim:int -> a:TensorLayout -> seq<int64>
Parameters
Type | Name | Description |
---|---|---|
int | dim | |
TensorLayout | a |
Returns
Type | Description |
---|---|
seq<int64> |