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

12 votes146 views1 answer
Python check namespace has instance
Python
0 vote119 views1 answer
Hello kitt
Python
85 votes120 views6 answers
Python sort dataframe by one column
Python
6 votes110 views2 answers
How to return the derivative of a function in python
Python
0 vote103 views1 answer
1036 solution python
Python
2 votes126 views2 answers
Python turn list of lists into list
Python
3 votes123 views2 answers
How to print answer 2 decimal places in python 3
Python
0 vote122 views1 answer
Amazing Trees with python turtle
Python
0 vote112 views1 answer
How to change kivy button size
Python
0 vote139 views2 answers
New window selenium python
Python