torch.broadcast_to — PyTorch 2.7 documentation (original) (raw)

• Docs >
• torch >
• torch.broadcast_to
• 

Shortcuts

torch.broadcast_to(input, shape) → Tensor¶

Broadcasts input to the shape shape. Equivalent to calling input.expand(shape). See expand() 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]])

• torch.broadcast_to
  • broadcast_to()