torch.broadcast_to¶
- torch.broadcast_to(input, shape) Tensor ¶
Broadcasts
input
to the shapeshape
. Equivalent to callinginput.expand(shape)
. Seeexpand()
for details.- Parameters
input (Tensor) – the input tensor.
shape (list, tuple, or
torch.Size
) – the new shape.
Example:
>>> x = torch.tensor([1, 2, 3]) >>> torch.broadcast_to(x, (3, 3)) tensor([[1, 2, 3], [1, 2, 3], [1, 2, 3]])