Rate this Page
★★★★★

torch.broadcast_to#

torch.broadcast_to(input,shape)→Tensor#

Broadcastsinput to the shapeshape.Equivalent to callinginput.expand(shape). Seeexpand() for details.

Parameters

  • input (Tensor) – the input tensor.

  • shape (list, tuple, ortorch.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]])