Method broadcastToSameInDims
val broadcastToSameInDims: seq<int> -> TensorLayout -> TensorLayout -> TensorLayout * TensorLayout
broadcasts to have the same size in the given dimensions
Declaration
val broadcastToSameInDims: dims:seq<int> -> ain:TensorLayout -> bin:TensorLayout -> TensorLayout * TensorLayout
Parameters
Type | Name | Description |
---|---|---|
seq<int> | dims | |
TensorLayout | ain | |
TensorLayout | bin |
Returns
Type | Description |
---|---|
TensorLayout * TensorLayout |