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 vote91 views1 answer
Nrf24l01 arduino raspberry pi
Python
0 vote146 views1 answer
Check if timestamp is NaT
Python
0 vote143 views1 answer
No pattern
Python
0 vote127 views1 answer
How to add column to heroku postgres in my django app
Python
1 vote139 views1 answer
How to add a cooment in python
Python
6 votes133 views2 answers
Python writeline file
Python
0 vote117 views1 answer
Add colorbar to 2d hist
Python
0 vote229 views1 answer
Inpuit inf terfminal ppython
Python
0 vote125 views1 answer
Mnist
Python
39 votes138 views6 answers
How to use a class in python
Python