oneflow.broadcast_like¶
-
oneflow.
broadcast_like
()¶ This operator broadcast tensor x to like_tensor according to the broadcast_axes.
- Parameters
- Returns
Broadcasted input Tensor.
- Return type
[Tensor]
For example:
>>> import oneflow as flow >>> x = flow.randn(3, 1, 1) >>> like_tensor = flow.randn(3, 4, 5) >>> broadcast_tensor = flow.broadcast_like(x, like_tensor, broadcast_axes=[1, 2]) >>> broadcast_tensor.shape oneflow.Size([3, 4, 5])