Skip to content

Slicing torch.Size returns tuple #730

@szagoruyko

Description

@szagoruyko

E.g. this: torch.Tensor(torch.Size((256,3,9,9,2))[:-1]) will return:

 256
   3
   9
   9
[torch.FloatTensor of size 4]

Slicing torch.Size should return torch.Size.

Metadata

Metadata

Assignees

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions