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