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

0 vote260 views1 answer
Explode function in python
Python
0 vote271 views1 answer
Id
Python
2 votes252 views2 answers
Del all variables python
Python
15 votes247 views3 answers
Remove punctuation python string library
Python
0 vote278 views1 answer
Scree plot sklearn
Python
0 vote272 views1 answer
How to repeat code in python until a condition is met
Python
0 vote309 views1 answer
Django python base 64 decode
Python Django
0 vote248 views1 answer
Featch detail of subscription in stripe api
Python
69 votes266 views4 answers
Python csv dictwriter
Python
6 votes285 views2 answers
Python check if any elements in a list are different
Python