Torch.tensor.expand

>>> x = torch.tensor([[1], [2], [3]])
>>> x.size()
torch.Size([3, 1])
>>> x.expand(3, 4)
tensor([[ 1,  1,  1,  1],
        [ 2,  2,  2,  2],
        [ 3,  3,  3,  3]])
>>> x.expand(-1, 4)   # -1 means not changing the size of that dimension
tensor([[ 1,  1,  1,  1],
        [ 2,  2,  2,  2],
        [ 3,  3,  3,  3]])

Source: pytorch.org

Similar codes

6 votes311 views2 answers
Iterable meaning
Python
0 vote330 views1 answer
Primary key auto increment python django
Python
11 votes350 views2 answers
Add pip to path
Python
0 vote386 views1 answer
Pyqt5 line edit password input
Python
2 votes315 views2 answers
Python replace multiple spacis with spaces
Python
0 vote328 views1 answer
Python print stderr
Python
7 votes385 views1 answer
If else python in single line
Python
6 votes323 views3 answers
Import pandas
Python
0 vote339 views1 answer
Draw bounding box matplotlib
Python
1 vote342 views2 answers
Python wget download
Python