1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
########## torch.float64/torch.int64/size=()+(4, 6)+(2, 3)+() ##########
# sparse meta tensor
tensor(crow_indices=tensor(..., size=(3,)),
col_indices=tensor(..., size=(0,)),
values=tensor(..., size=(0, 2, 3)), device='meta', size=(4, 6),
dtype=torch.float64, layout=torch.sparse_bsr)
########## torch.float64/torch.int64/size=()+(4, 6)+(2, 3)+(3,) ##########
# sparse meta tensor
tensor(crow_indices=tensor(..., size=(3,)),
col_indices=tensor(..., size=(0,)),
values=tensor(..., size=(0, 2, 3, 3)), device='meta', size=(4, 6, 3),
dtype=torch.float64, layout=torch.sparse_bsr)
########## torch.float64/torch.int64/size=(2,)+(4, 6)+(2, 3)+() ##########
# sparse meta tensor
tensor(crow_indices=tensor(..., size=(2, 3)),
col_indices=tensor(..., size=(2, 0)),
values=tensor(..., size=(2, 0, 2, 3)), device='meta', size=(2, 4, 6),
dtype=torch.float64, layout=torch.sparse_bsr)
########## torch.float64/torch.int64/size=(2,)+(4, 6)+(2, 3)+(3,) ##########
# sparse meta tensor
tensor(crow_indices=tensor(..., size=(2, 3)),
col_indices=tensor(..., size=(2, 0)),
values=tensor(..., size=(2, 0, 2, 3, 3)), device='meta',
size=(2, 4, 6, 3), dtype=torch.float64, layout=torch.sparse_bsr)
|
