Method broadcastToShape
val broadcastToShape: int64 list -> TensorLayout -> TensorLayout
broadcasts a tensor to the given shape
Declaration
val broadcastToShape: bs:int64 list -> ain:TensorLayout -> TensorLayoutParameters
| Type | Name | Description |
|---|---|---|
| int64 list | bs | |
| TensorLayout | ain |
Returns
| Type | Description |
|---|---|
| TensorLayout |