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 |