Breaking News: Grepper is joining You.com. Read the official announcement!

pytorch flatten tensor

Fancy Flamingo answered on March 7, 2024 Popularity 3/10 Helpfulness 2/10

answer pytorch flatten tensor

pytorch flatten tensor

Comment

torch.flatten

>>> t = torch.tensor([[[1, 2],
...                    [3, 4]],
...                   [[5, 6],
...                    [7, 8]]])
>>> torch.flatten(t)
tensor([1, 2, 3, 4, 5, 6, 7, 8])
>>> torch.flatten(t, start_dim=1)
tensor([[1, 2, 3, 4],
        [5, 6, 7, 8]])

https://pytorch.org/docs/stable/generated/torch.flatten.html

Popularity 3/10 Helpfulness 2/10 Language python

Source: Grepper

Tags: flatten python pytorch tensor

Link to this answer
Share Copy Link