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)+()+() ##########
# sparse meta tensor
tensor(crow_indices=tensor(..., size=(5,)),
col_indices=tensor(..., size=(0,)),
values=tensor(..., size=(0,)), device='meta', size=(4, 6),
dtype=torch.float64, layout=torch.sparse_csr)
########## torch.float64/torch.int64/size=()+(4, 6)+()+(3,) ##########
# sparse meta tensor
tensor(crow_indices=tensor(..., size=(5,)),
col_indices=tensor(..., size=(0,)),
values=tensor(..., size=(0, 3)), device='meta', size=(4, 6, 3),
dtype=torch.float64, layout=torch.sparse_csr)
########## torch.float64/torch.int64/size=(2,)+(4, 6)+()+() ##########
# sparse meta tensor
tensor(crow_indices=tensor(..., size=(2, 5)),
col_indices=tensor(..., size=(2, 0)),
values=tensor(..., size=(2, 0)), device='meta', size=(2, 4, 6),
dtype=torch.float64, layout=torch.sparse_csr)
########## torch.float64/torch.int64/size=(2,)+(4, 6)+()+(3,) ##########
# sparse meta tensor
tensor(crow_indices=tensor(..., size=(2, 5)),
col_indices=tensor(..., size=(2, 0)),
values=tensor(..., size=(2, 0, 3)), device='meta', size=(2, 4, 6, 3),
dtype=torch.float64, layout=torch.sparse_csr)
|