Method broadcastDim
val broadcastDim: int -> int64 -> TensorLayout -> TensorLayout
broadcast the given dimension to the given size
Declaration
val broadcastDim: dim:int -> size:int64 -> a:TensorLayout -> TensorLayout
Parameters
| Type | Name | Description |
|---|---|---|
| int | dim | |
| int64 | size | |
| TensorLayout | a |
Returns
| Type | Description |
|---|---|
| TensorLayout |