oneflow.broadcast_like

oneflow.broadcast_like()

This operator broadcast tensor x to like_tensor according to the broadcast_axes.

Parameters
  • x (Tensor) – The input Tensor.

  • like_tensor (Tensor) – The like Tensor.

  • broadcast_axes (Optional[Sequence], optional) – The axes you want to broadcast. Defaults to None.

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])