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