| ########## torch.float32/torch.int32/size=()+(4, 3)+() ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([0, 2, 4]), |
| col_indices=tensor([0, 1, 0, 2]), |
| values=tensor([[[1.], |
| [2.]], |
| |
| [[2.], |
| [3.]], |
| |
| [[3.], |
| [4.]], |
| |
| [[4.], |
| [5.]]]), device='cuda:0', size=(4, 3), nnz=4, |
| layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([0, 2, 4], device='cuda:0', dtype=torch.int32) |
| # _col_indices |
| tensor([0, 1, 0, 2], device='cuda:0', dtype=torch.int32) |
| # _values |
| tensor([[[1.], |
| [2.]], |
| |
| [[2.], |
| [3.]], |
| |
| [[3.], |
| [4.]], |
| |
| [[4.], |
| [5.]]], device='cuda:0') |
| |
| ########## torch.float32/torch.int32/size=()+(0, 0)+() ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([0]), |
| col_indices=tensor([], size=(0,)), |
| values=tensor([], size=(0, 2, 1)), device='cuda:0', size=(0, 0), nnz=0, |
| layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([0], device='cuda:0', dtype=torch.int32) |
| # _col_indices |
| tensor([], device='cuda:0', dtype=torch.int32) |
| # _values |
| tensor([], device='cuda:0', size=(0, 2, 1)) |
| |
| ########## torch.float32/torch.int32/size=(2,)+(2, 6)+() ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([[0, 2, 4], |
| [0, 3, 4]]), |
| col_indices=tensor([[0, 1, 0, 1], |
| [0, 1, 2, 0]]), |
| values=tensor([[[[ 1., 11.]], |
| |
| [[ 2., 12.]], |
| |
| [[ 3., 13.]], |
| |
| [[ 4., 14.]]], |
| |
| |
| [[[ 5., 15.]], |
| |
| [[ 6., 16.]], |
| |
| [[ 7., 17.]], |
| |
| [[ 8., 18.]]]]), device='cuda:0', size=(2, 2, 6), nnz=4, |
| layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([[0, 2, 4], |
| [0, 3, 4]], device='cuda:0', dtype=torch.int32) |
| # _col_indices |
| tensor([[0, 1, 0, 1], |
| [0, 1, 2, 0]], device='cuda:0', dtype=torch.int32) |
| # _values |
| tensor([[[[ 1., 11.]], |
| |
| [[ 2., 12.]], |
| |
| [[ 3., 13.]], |
| |
| [[ 4., 14.]]], |
| |
| |
| [[[ 5., 15.]], |
| |
| [[ 6., 16.]], |
| |
| [[ 7., 17.]], |
| |
| [[ 8., 18.]]]], device='cuda:0') |
| |
| ########## torch.float32/torch.int32/size=(2, 3)+(4, 9)+() ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([[[0, 2, 4], |
| [0, 3, 4], |
| [0, 1, 4]], |
| |
| [[0, 1, 4], |
| [0, 2, 4], |
| [0, 3, 4]]]), |
| col_indices=tensor([[[0, 1, 0, 1], |
| [0, 1, 2, 0], |
| [0, 0, 1, 2]], |
| |
| [[1, 0, 1, 2], |
| [0, 2, 0, 1], |
| [0, 1, 2, 1]]]), |
| values=tensor([[[[[ 1., 11., 21.], |
| [ 2., 12., 22.]], |
| |
| [[ 2., 12., 22.], |
| [ 3., 13., 23.]], |
| |
| [[ 3., 13., 23.], |
| [ 4., 14., 24.]], |
| |
| [[ 4., 14., 24.], |
| [ 5., 15., 25.]]], |
| |
| |
| [[[ 5., 15., 25.], |
| [ 6., 16., 26.]], |
| |
| [[ 6., 16., 26.], |
| [ 7., 17., 27.]], |
| |
| [[ 7., 17., 27.], |
| [ 8., 18., 28.]], |
| |
| [[ 8., 18., 28.], |
| [ 9., 19., 29.]]], |
| |
| |
| [[[ 9., 19., 29.], |
| [10., 20., 30.]], |
| |
| [[10., 20., 30.], |
| [11., 21., 31.]], |
| |
| [[11., 21., 31.], |
| [12., 22., 32.]], |
| |
| [[12., 22., 32.], |
| [13., 23., 33.]]]], |
| |
| |
| |
| [[[[13., 23., 33.], |
| [14., 24., 34.]], |
| |
| [[14., 24., 34.], |
| [15., 25., 35.]], |
| |
| [[15., 25., 35.], |
| [16., 26., 36.]], |
| |
| [[16., 26., 36.], |
| [17., 27., 37.]]], |
| |
| |
| [[[17., 27., 37.], |
| [18., 28., 38.]], |
| |
| [[18., 28., 38.], |
| [19., 29., 39.]], |
| |
| [[19., 29., 39.], |
| [20., 30., 40.]], |
| |
| [[20., 30., 40.], |
| [21., 31., 41.]]], |
| |
| |
| [[[21., 31., 41.], |
| [22., 32., 42.]], |
| |
| [[22., 32., 42.], |
| [23., 33., 43.]], |
| |
| [[23., 33., 43.], |
| [24., 34., 44.]], |
| |
| [[24., 34., 44.], |
| [25., 35., 45.]]]]]), device='cuda:0', |
| size=(2, 3, 4, 9), nnz=4, layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([[[0, 2, 4], |
| [0, 3, 4], |
| [0, 1, 4]], |
| |
| [[0, 1, 4], |
| [0, 2, 4], |
| [0, 3, 4]]], device='cuda:0', dtype=torch.int32) |
| # _col_indices |
| tensor([[[0, 1, 0, 1], |
| [0, 1, 2, 0], |
| [0, 0, 1, 2]], |
| |
| [[1, 0, 1, 2], |
| [0, 2, 0, 1], |
| [0, 1, 2, 1]]], device='cuda:0', dtype=torch.int32) |
| # _values |
| tensor([[[[[ 1., 11., 21.], |
| [ 2., 12., 22.]], |
| |
| [[ 2., 12., 22.], |
| [ 3., 13., 23.]], |
| |
| [[ 3., 13., 23.], |
| [ 4., 14., 24.]], |
| |
| [[ 4., 14., 24.], |
| [ 5., 15., 25.]]], |
| |
| |
| [[[ 5., 15., 25.], |
| [ 6., 16., 26.]], |
| |
| [[ 6., 16., 26.], |
| [ 7., 17., 27.]], |
| |
| [[ 7., 17., 27.], |
| [ 8., 18., 28.]], |
| |
| [[ 8., 18., 28.], |
| [ 9., 19., 29.]]], |
| |
| |
| [[[ 9., 19., 29.], |
| [10., 20., 30.]], |
| |
| [[10., 20., 30.], |
| [11., 21., 31.]], |
| |
| [[11., 21., 31.], |
| [12., 22., 32.]], |
| |
| [[12., 22., 32.], |
| [13., 23., 33.]]]], |
| |
| |
| |
| [[[[13., 23., 33.], |
| [14., 24., 34.]], |
| |
| [[14., 24., 34.], |
| [15., 25., 35.]], |
| |
| [[15., 25., 35.], |
| [16., 26., 36.]], |
| |
| [[16., 26., 36.], |
| [17., 27., 37.]]], |
| |
| |
| [[[17., 27., 37.], |
| [18., 28., 38.]], |
| |
| [[18., 28., 38.], |
| [19., 29., 39.]], |
| |
| [[19., 29., 39.], |
| [20., 30., 40.]], |
| |
| [[20., 30., 40.], |
| [21., 31., 41.]]], |
| |
| |
| [[[21., 31., 41.], |
| [22., 32., 42.]], |
| |
| [[22., 32., 42.], |
| [23., 33., 43.]], |
| |
| [[23., 33., 43.], |
| [24., 34., 44.]], |
| |
| [[24., 34., 44.], |
| [25., 35., 45.]]]]], device='cuda:0') |
| |
| |
| ########## torch.float64/torch.int32/size=()+(4, 3)+() ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([0, 2, 4]), |
| col_indices=tensor([0, 1, 0, 2]), |
| values=tensor([[[1.], |
| [2.]], |
| |
| [[2.], |
| [3.]], |
| |
| [[3.], |
| [4.]], |
| |
| [[4.], |
| [5.]]]), device='cuda:0', size=(4, 3), nnz=4, |
| dtype=torch.float64, layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([0, 2, 4], device='cuda:0', dtype=torch.int32) |
| # _col_indices |
| tensor([0, 1, 0, 2], device='cuda:0', dtype=torch.int32) |
| # _values |
| tensor([[[1.], |
| [2.]], |
| |
| [[2.], |
| [3.]], |
| |
| [[3.], |
| [4.]], |
| |
| [[4.], |
| [5.]]], device='cuda:0', dtype=torch.float64) |
| |
| ########## torch.float64/torch.int32/size=()+(0, 0)+() ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([0]), |
| col_indices=tensor([], size=(0,)), |
| values=tensor([], size=(0, 2, 1)), device='cuda:0', size=(0, 0), nnz=0, |
| dtype=torch.float64, layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([0], device='cuda:0', dtype=torch.int32) |
| # _col_indices |
| tensor([], device='cuda:0', dtype=torch.int32) |
| # _values |
| tensor([], device='cuda:0', size=(0, 2, 1), dtype=torch.float64) |
| |
| ########## torch.float64/torch.int32/size=(2,)+(2, 6)+() ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([[0, 2, 4], |
| [0, 3, 4]]), |
| col_indices=tensor([[0, 1, 0, 1], |
| [0, 1, 2, 0]]), |
| values=tensor([[[[ 1., 11.]], |
| |
| [[ 2., 12.]], |
| |
| [[ 3., 13.]], |
| |
| [[ 4., 14.]]], |
| |
| |
| [[[ 5., 15.]], |
| |
| [[ 6., 16.]], |
| |
| [[ 7., 17.]], |
| |
| [[ 8., 18.]]]]), device='cuda:0', size=(2, 2, 6), nnz=4, |
| dtype=torch.float64, layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([[0, 2, 4], |
| [0, 3, 4]], device='cuda:0', dtype=torch.int32) |
| # _col_indices |
| tensor([[0, 1, 0, 1], |
| [0, 1, 2, 0]], device='cuda:0', dtype=torch.int32) |
| # _values |
| tensor([[[[ 1., 11.]], |
| |
| [[ 2., 12.]], |
| |
| [[ 3., 13.]], |
| |
| [[ 4., 14.]]], |
| |
| |
| [[[ 5., 15.]], |
| |
| [[ 6., 16.]], |
| |
| [[ 7., 17.]], |
| |
| [[ 8., 18.]]]], device='cuda:0', dtype=torch.float64) |
| |
| ########## torch.float64/torch.int32/size=(2, 3)+(4, 9)+() ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([[[0, 2, 4], |
| [0, 3, 4], |
| [0, 1, 4]], |
| |
| [[0, 1, 4], |
| [0, 2, 4], |
| [0, 3, 4]]]), |
| col_indices=tensor([[[0, 1, 0, 1], |
| [0, 1, 2, 0], |
| [0, 0, 1, 2]], |
| |
| [[1, 0, 1, 2], |
| [0, 2, 0, 1], |
| [0, 1, 2, 1]]]), |
| values=tensor([[[[[ 1., 11., 21.], |
| [ 2., 12., 22.]], |
| |
| [[ 2., 12., 22.], |
| [ 3., 13., 23.]], |
| |
| [[ 3., 13., 23.], |
| [ 4., 14., 24.]], |
| |
| [[ 4., 14., 24.], |
| [ 5., 15., 25.]]], |
| |
| |
| [[[ 5., 15., 25.], |
| [ 6., 16., 26.]], |
| |
| [[ 6., 16., 26.], |
| [ 7., 17., 27.]], |
| |
| [[ 7., 17., 27.], |
| [ 8., 18., 28.]], |
| |
| [[ 8., 18., 28.], |
| [ 9., 19., 29.]]], |
| |
| |
| [[[ 9., 19., 29.], |
| [10., 20., 30.]], |
| |
| [[10., 20., 30.], |
| [11., 21., 31.]], |
| |
| [[11., 21., 31.], |
| [12., 22., 32.]], |
| |
| [[12., 22., 32.], |
| [13., 23., 33.]]]], |
| |
| |
| |
| [[[[13., 23., 33.], |
| [14., 24., 34.]], |
| |
| [[14., 24., 34.], |
| [15., 25., 35.]], |
| |
| [[15., 25., 35.], |
| [16., 26., 36.]], |
| |
| [[16., 26., 36.], |
| [17., 27., 37.]]], |
| |
| |
| [[[17., 27., 37.], |
| [18., 28., 38.]], |
| |
| [[18., 28., 38.], |
| [19., 29., 39.]], |
| |
| [[19., 29., 39.], |
| [20., 30., 40.]], |
| |
| [[20., 30., 40.], |
| [21., 31., 41.]]], |
| |
| |
| [[[21., 31., 41.], |
| [22., 32., 42.]], |
| |
| [[22., 32., 42.], |
| [23., 33., 43.]], |
| |
| [[23., 33., 43.], |
| [24., 34., 44.]], |
| |
| [[24., 34., 44.], |
| [25., 35., 45.]]]]]), device='cuda:0', |
| size=(2, 3, 4, 9), nnz=4, dtype=torch.float64, layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([[[0, 2, 4], |
| [0, 3, 4], |
| [0, 1, 4]], |
| |
| [[0, 1, 4], |
| [0, 2, 4], |
| [0, 3, 4]]], device='cuda:0', dtype=torch.int32) |
| # _col_indices |
| tensor([[[0, 1, 0, 1], |
| [0, 1, 2, 0], |
| [0, 0, 1, 2]], |
| |
| [[1, 0, 1, 2], |
| [0, 2, 0, 1], |
| [0, 1, 2, 1]]], device='cuda:0', dtype=torch.int32) |
| # _values |
| tensor([[[[[ 1., 11., 21.], |
| [ 2., 12., 22.]], |
| |
| [[ 2., 12., 22.], |
| [ 3., 13., 23.]], |
| |
| [[ 3., 13., 23.], |
| [ 4., 14., 24.]], |
| |
| [[ 4., 14., 24.], |
| [ 5., 15., 25.]]], |
| |
| |
| [[[ 5., 15., 25.], |
| [ 6., 16., 26.]], |
| |
| [[ 6., 16., 26.], |
| [ 7., 17., 27.]], |
| |
| [[ 7., 17., 27.], |
| [ 8., 18., 28.]], |
| |
| [[ 8., 18., 28.], |
| [ 9., 19., 29.]]], |
| |
| |
| [[[ 9., 19., 29.], |
| [10., 20., 30.]], |
| |
| [[10., 20., 30.], |
| [11., 21., 31.]], |
| |
| [[11., 21., 31.], |
| [12., 22., 32.]], |
| |
| [[12., 22., 32.], |
| [13., 23., 33.]]]], |
| |
| |
| |
| [[[[13., 23., 33.], |
| [14., 24., 34.]], |
| |
| [[14., 24., 34.], |
| [15., 25., 35.]], |
| |
| [[15., 25., 35.], |
| [16., 26., 36.]], |
| |
| [[16., 26., 36.], |
| [17., 27., 37.]]], |
| |
| |
| [[[17., 27., 37.], |
| [18., 28., 38.]], |
| |
| [[18., 28., 38.], |
| [19., 29., 39.]], |
| |
| [[19., 29., 39.], |
| [20., 30., 40.]], |
| |
| [[20., 30., 40.], |
| [21., 31., 41.]]], |
| |
| |
| [[[21., 31., 41.], |
| [22., 32., 42.]], |
| |
| [[22., 32., 42.], |
| [23., 33., 43.]], |
| |
| [[23., 33., 43.], |
| [24., 34., 44.]], |
| |
| [[24., 34., 44.], |
| [25., 35., 45.]]]]], device='cuda:0', dtype=torch.float64) |
| |
| |
| ########## torch.float32/torch.int64/size=()+(4, 3)+() ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([0, 2, 4]), |
| col_indices=tensor([0, 1, 0, 2]), |
| values=tensor([[[1.], |
| [2.]], |
| |
| [[2.], |
| [3.]], |
| |
| [[3.], |
| [4.]], |
| |
| [[4.], |
| [5.]]]), device='cuda:0', size=(4, 3), nnz=4, |
| layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([0, 2, 4], device='cuda:0') |
| # _col_indices |
| tensor([0, 1, 0, 2], device='cuda:0') |
| # _values |
| tensor([[[1.], |
| [2.]], |
| |
| [[2.], |
| [3.]], |
| |
| [[3.], |
| [4.]], |
| |
| [[4.], |
| [5.]]], device='cuda:0') |
| |
| ########## torch.float32/torch.int64/size=()+(0, 0)+() ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([0]), |
| col_indices=tensor([], size=(0,)), |
| values=tensor([], size=(0, 2, 1)), device='cuda:0', size=(0, 0), nnz=0, |
| layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([0], device='cuda:0') |
| # _col_indices |
| tensor([], device='cuda:0', dtype=torch.int64) |
| # _values |
| tensor([], device='cuda:0', size=(0, 2, 1)) |
| |
| ########## torch.float32/torch.int64/size=(2,)+(2, 6)+() ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([[0, 2, 4], |
| [0, 3, 4]]), |
| col_indices=tensor([[0, 1, 0, 1], |
| [0, 1, 2, 0]]), |
| values=tensor([[[[ 1., 11.]], |
| |
| [[ 2., 12.]], |
| |
| [[ 3., 13.]], |
| |
| [[ 4., 14.]]], |
| |
| |
| [[[ 5., 15.]], |
| |
| [[ 6., 16.]], |
| |
| [[ 7., 17.]], |
| |
| [[ 8., 18.]]]]), device='cuda:0', size=(2, 2, 6), nnz=4, |
| layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([[0, 2, 4], |
| [0, 3, 4]], device='cuda:0') |
| # _col_indices |
| tensor([[0, 1, 0, 1], |
| [0, 1, 2, 0]], device='cuda:0') |
| # _values |
| tensor([[[[ 1., 11.]], |
| |
| [[ 2., 12.]], |
| |
| [[ 3., 13.]], |
| |
| [[ 4., 14.]]], |
| |
| |
| [[[ 5., 15.]], |
| |
| [[ 6., 16.]], |
| |
| [[ 7., 17.]], |
| |
| [[ 8., 18.]]]], device='cuda:0') |
| |
| ########## torch.float32/torch.int64/size=(2, 3)+(4, 9)+() ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([[[0, 2, 4], |
| [0, 3, 4], |
| [0, 1, 4]], |
| |
| [[0, 1, 4], |
| [0, 2, 4], |
| [0, 3, 4]]]), |
| col_indices=tensor([[[0, 1, 0, 1], |
| [0, 1, 2, 0], |
| [0, 0, 1, 2]], |
| |
| [[1, 0, 1, 2], |
| [0, 2, 0, 1], |
| [0, 1, 2, 1]]]), |
| values=tensor([[[[[ 1., 11., 21.], |
| [ 2., 12., 22.]], |
| |
| [[ 2., 12., 22.], |
| [ 3., 13., 23.]], |
| |
| [[ 3., 13., 23.], |
| [ 4., 14., 24.]], |
| |
| [[ 4., 14., 24.], |
| [ 5., 15., 25.]]], |
| |
| |
| [[[ 5., 15., 25.], |
| [ 6., 16., 26.]], |
| |
| [[ 6., 16., 26.], |
| [ 7., 17., 27.]], |
| |
| [[ 7., 17., 27.], |
| [ 8., 18., 28.]], |
| |
| [[ 8., 18., 28.], |
| [ 9., 19., 29.]]], |
| |
| |
| [[[ 9., 19., 29.], |
| [10., 20., 30.]], |
| |
| [[10., 20., 30.], |
| [11., 21., 31.]], |
| |
| [[11., 21., 31.], |
| [12., 22., 32.]], |
| |
| [[12., 22., 32.], |
| [13., 23., 33.]]]], |
| |
| |
| |
| [[[[13., 23., 33.], |
| [14., 24., 34.]], |
| |
| [[14., 24., 34.], |
| [15., 25., 35.]], |
| |
| [[15., 25., 35.], |
| [16., 26., 36.]], |
| |
| [[16., 26., 36.], |
| [17., 27., 37.]]], |
| |
| |
| [[[17., 27., 37.], |
| [18., 28., 38.]], |
| |
| [[18., 28., 38.], |
| [19., 29., 39.]], |
| |
| [[19., 29., 39.], |
| [20., 30., 40.]], |
| |
| [[20., 30., 40.], |
| [21., 31., 41.]]], |
| |
| |
| [[[21., 31., 41.], |
| [22., 32., 42.]], |
| |
| [[22., 32., 42.], |
| [23., 33., 43.]], |
| |
| [[23., 33., 43.], |
| [24., 34., 44.]], |
| |
| [[24., 34., 44.], |
| [25., 35., 45.]]]]]), device='cuda:0', |
| size=(2, 3, 4, 9), nnz=4, layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([[[0, 2, 4], |
| [0, 3, 4], |
| [0, 1, 4]], |
| |
| [[0, 1, 4], |
| [0, 2, 4], |
| [0, 3, 4]]], device='cuda:0') |
| # _col_indices |
| tensor([[[0, 1, 0, 1], |
| [0, 1, 2, 0], |
| [0, 0, 1, 2]], |
| |
| [[1, 0, 1, 2], |
| [0, 2, 0, 1], |
| [0, 1, 2, 1]]], device='cuda:0') |
| # _values |
| tensor([[[[[ 1., 11., 21.], |
| [ 2., 12., 22.]], |
| |
| [[ 2., 12., 22.], |
| [ 3., 13., 23.]], |
| |
| [[ 3., 13., 23.], |
| [ 4., 14., 24.]], |
| |
| [[ 4., 14., 24.], |
| [ 5., 15., 25.]]], |
| |
| |
| [[[ 5., 15., 25.], |
| [ 6., 16., 26.]], |
| |
| [[ 6., 16., 26.], |
| [ 7., 17., 27.]], |
| |
| [[ 7., 17., 27.], |
| [ 8., 18., 28.]], |
| |
| [[ 8., 18., 28.], |
| [ 9., 19., 29.]]], |
| |
| |
| [[[ 9., 19., 29.], |
| [10., 20., 30.]], |
| |
| [[10., 20., 30.], |
| [11., 21., 31.]], |
| |
| [[11., 21., 31.], |
| [12., 22., 32.]], |
| |
| [[12., 22., 32.], |
| [13., 23., 33.]]]], |
| |
| |
| |
| [[[[13., 23., 33.], |
| [14., 24., 34.]], |
| |
| [[14., 24., 34.], |
| [15., 25., 35.]], |
| |
| [[15., 25., 35.], |
| [16., 26., 36.]], |
| |
| [[16., 26., 36.], |
| [17., 27., 37.]]], |
| |
| |
| [[[17., 27., 37.], |
| [18., 28., 38.]], |
| |
| [[18., 28., 38.], |
| [19., 29., 39.]], |
| |
| [[19., 29., 39.], |
| [20., 30., 40.]], |
| |
| [[20., 30., 40.], |
| [21., 31., 41.]]], |
| |
| |
| [[[21., 31., 41.], |
| [22., 32., 42.]], |
| |
| [[22., 32., 42.], |
| [23., 33., 43.]], |
| |
| [[23., 33., 43.], |
| [24., 34., 44.]], |
| |
| [[24., 34., 44.], |
| [25., 35., 45.]]]]], device='cuda:0') |
| |
| |
| ########## torch.float64/torch.int64/size=()+(4, 3)+() ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([0, 2, 4]), |
| col_indices=tensor([0, 1, 0, 2]), |
| values=tensor([[[1.], |
| [2.]], |
| |
| [[2.], |
| [3.]], |
| |
| [[3.], |
| [4.]], |
| |
| [[4.], |
| [5.]]]), device='cuda:0', size=(4, 3), nnz=4, |
| dtype=torch.float64, layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([0, 2, 4], device='cuda:0') |
| # _col_indices |
| tensor([0, 1, 0, 2], device='cuda:0') |
| # _values |
| tensor([[[1.], |
| [2.]], |
| |
| [[2.], |
| [3.]], |
| |
| [[3.], |
| [4.]], |
| |
| [[4.], |
| [5.]]], device='cuda:0', dtype=torch.float64) |
| |
| ########## torch.float64/torch.int64/size=()+(0, 0)+() ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([0]), |
| col_indices=tensor([], size=(0,)), |
| values=tensor([], size=(0, 2, 1)), device='cuda:0', size=(0, 0), nnz=0, |
| dtype=torch.float64, layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([0], device='cuda:0') |
| # _col_indices |
| tensor([], device='cuda:0', dtype=torch.int64) |
| # _values |
| tensor([], device='cuda:0', size=(0, 2, 1), dtype=torch.float64) |
| |
| ########## torch.float64/torch.int64/size=(2,)+(2, 6)+() ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([[0, 2, 4], |
| [0, 3, 4]]), |
| col_indices=tensor([[0, 1, 0, 1], |
| [0, 1, 2, 0]]), |
| values=tensor([[[[ 1., 11.]], |
| |
| [[ 2., 12.]], |
| |
| [[ 3., 13.]], |
| |
| [[ 4., 14.]]], |
| |
| |
| [[[ 5., 15.]], |
| |
| [[ 6., 16.]], |
| |
| [[ 7., 17.]], |
| |
| [[ 8., 18.]]]]), device='cuda:0', size=(2, 2, 6), nnz=4, |
| dtype=torch.float64, layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([[0, 2, 4], |
| [0, 3, 4]], device='cuda:0') |
| # _col_indices |
| tensor([[0, 1, 0, 1], |
| [0, 1, 2, 0]], device='cuda:0') |
| # _values |
| tensor([[[[ 1., 11.]], |
| |
| [[ 2., 12.]], |
| |
| [[ 3., 13.]], |
| |
| [[ 4., 14.]]], |
| |
| |
| [[[ 5., 15.]], |
| |
| [[ 6., 16.]], |
| |
| [[ 7., 17.]], |
| |
| [[ 8., 18.]]]], device='cuda:0', dtype=torch.float64) |
| |
| ########## torch.float64/torch.int64/size=(2, 3)+(4, 9)+() ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([[[0, 2, 4], |
| [0, 3, 4], |
| [0, 1, 4]], |
| |
| [[0, 1, 4], |
| [0, 2, 4], |
| [0, 3, 4]]]), |
| col_indices=tensor([[[0, 1, 0, 1], |
| [0, 1, 2, 0], |
| [0, 0, 1, 2]], |
| |
| [[1, 0, 1, 2], |
| [0, 2, 0, 1], |
| [0, 1, 2, 1]]]), |
| values=tensor([[[[[ 1., 11., 21.], |
| [ 2., 12., 22.]], |
| |
| [[ 2., 12., 22.], |
| [ 3., 13., 23.]], |
| |
| [[ 3., 13., 23.], |
| [ 4., 14., 24.]], |
| |
| [[ 4., 14., 24.], |
| [ 5., 15., 25.]]], |
| |
| |
| [[[ 5., 15., 25.], |
| [ 6., 16., 26.]], |
| |
| [[ 6., 16., 26.], |
| [ 7., 17., 27.]], |
| |
| [[ 7., 17., 27.], |
| [ 8., 18., 28.]], |
| |
| [[ 8., 18., 28.], |
| [ 9., 19., 29.]]], |
| |
| |
| [[[ 9., 19., 29.], |
| [10., 20., 30.]], |
| |
| [[10., 20., 30.], |
| [11., 21., 31.]], |
| |
| [[11., 21., 31.], |
| [12., 22., 32.]], |
| |
| [[12., 22., 32.], |
| [13., 23., 33.]]]], |
| |
| |
| |
| [[[[13., 23., 33.], |
| [14., 24., 34.]], |
| |
| [[14., 24., 34.], |
| [15., 25., 35.]], |
| |
| [[15., 25., 35.], |
| [16., 26., 36.]], |
| |
| [[16., 26., 36.], |
| [17., 27., 37.]]], |
| |
| |
| [[[17., 27., 37.], |
| [18., 28., 38.]], |
| |
| [[18., 28., 38.], |
| [19., 29., 39.]], |
| |
| [[19., 29., 39.], |
| [20., 30., 40.]], |
| |
| [[20., 30., 40.], |
| [21., 31., 41.]]], |
| |
| |
| [[[21., 31., 41.], |
| [22., 32., 42.]], |
| |
| [[22., 32., 42.], |
| [23., 33., 43.]], |
| |
| [[23., 33., 43.], |
| [24., 34., 44.]], |
| |
| [[24., 34., 44.], |
| [25., 35., 45.]]]]]), device='cuda:0', |
| size=(2, 3, 4, 9), nnz=4, dtype=torch.float64, layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([[[0, 2, 4], |
| [0, 3, 4], |
| [0, 1, 4]], |
| |
| [[0, 1, 4], |
| [0, 2, 4], |
| [0, 3, 4]]], device='cuda:0') |
| # _col_indices |
| tensor([[[0, 1, 0, 1], |
| [0, 1, 2, 0], |
| [0, 0, 1, 2]], |
| |
| [[1, 0, 1, 2], |
| [0, 2, 0, 1], |
| [0, 1, 2, 1]]], device='cuda:0') |
| # _values |
| tensor([[[[[ 1., 11., 21.], |
| [ 2., 12., 22.]], |
| |
| [[ 2., 12., 22.], |
| [ 3., 13., 23.]], |
| |
| [[ 3., 13., 23.], |
| [ 4., 14., 24.]], |
| |
| [[ 4., 14., 24.], |
| [ 5., 15., 25.]]], |
| |
| |
| [[[ 5., 15., 25.], |
| [ 6., 16., 26.]], |
| |
| [[ 6., 16., 26.], |
| [ 7., 17., 27.]], |
| |
| [[ 7., 17., 27.], |
| [ 8., 18., 28.]], |
| |
| [[ 8., 18., 28.], |
| [ 9., 19., 29.]]], |
| |
| |
| [[[ 9., 19., 29.], |
| [10., 20., 30.]], |
| |
| [[10., 20., 30.], |
| [11., 21., 31.]], |
| |
| [[11., 21., 31.], |
| [12., 22., 32.]], |
| |
| [[12., 22., 32.], |
| [13., 23., 33.]]]], |
| |
| |
| |
| [[[[13., 23., 33.], |
| [14., 24., 34.]], |
| |
| [[14., 24., 34.], |
| [15., 25., 35.]], |
| |
| [[15., 25., 35.], |
| [16., 26., 36.]], |
| |
| [[16., 26., 36.], |
| [17., 27., 37.]]], |
| |
| |
| [[[17., 27., 37.], |
| [18., 28., 38.]], |
| |
| [[18., 28., 38.], |
| [19., 29., 39.]], |
| |
| [[19., 29., 39.], |
| [20., 30., 40.]], |
| |
| [[20., 30., 40.], |
| [21., 31., 41.]]], |
| |
| |
| [[[21., 31., 41.], |
| [22., 32., 42.]], |
| |
| [[22., 32., 42.], |
| [23., 33., 43.]], |
| |
| [[23., 33., 43.], |
| [24., 34., 44.]], |
| |
| [[24., 34., 44.], |
| [25., 35., 45.]]]]], device='cuda:0', dtype=torch.float64) |
| |
| |
| ########## torch.float32/torch.int32/size=()+(6, 6)+(2,) ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([0, 2, 4]), |
| col_indices=tensor([0, 1, 0, 2]), |
| values=tensor([[[[ 1., 101.], |
| [ 11., 111.]], |
| |
| [[ 2., 102.], |
| [ 12., 112.]], |
| |
| [[ 3., 103.], |
| [ 13., 113.]]], |
| |
| |
| [[[ 2., 102.], |
| [ 12., 112.]], |
| |
| [[ 3., 103.], |
| [ 13., 113.]], |
| |
| [[ 4., 104.], |
| [ 14., 114.]]], |
| |
| |
| [[[ 3., 103.], |
| [ 13., 113.]], |
| |
| [[ 4., 104.], |
| [ 14., 114.]], |
| |
| [[ 5., 105.], |
| [ 15., 115.]]], |
| |
| |
| [[[ 4., 104.], |
| [ 14., 114.]], |
| |
| [[ 5., 105.], |
| [ 15., 115.]], |
| |
| [[ 6., 106.], |
| [ 16., 116.]]]]), device='cuda:0', size=(6, 6, 2), |
| nnz=4, layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([0, 2, 4], device='cuda:0', dtype=torch.int32) |
| # _col_indices |
| tensor([0, 1, 0, 2], device='cuda:0', dtype=torch.int32) |
| # _values |
| tensor([[[[ 1., 101.], |
| [ 11., 111.]], |
| |
| [[ 2., 102.], |
| [ 12., 112.]], |
| |
| [[ 3., 103.], |
| [ 13., 113.]]], |
| |
| |
| [[[ 2., 102.], |
| [ 12., 112.]], |
| |
| [[ 3., 103.], |
| [ 13., 113.]], |
| |
| [[ 4., 104.], |
| [ 14., 114.]]], |
| |
| |
| [[[ 3., 103.], |
| [ 13., 113.]], |
| |
| [[ 4., 104.], |
| [ 14., 114.]], |
| |
| [[ 5., 105.], |
| [ 15., 115.]]], |
| |
| |
| [[[ 4., 104.], |
| [ 14., 114.]], |
| |
| [[ 5., 105.], |
| [ 15., 115.]], |
| |
| [[ 6., 106.], |
| [ 16., 116.]]]], device='cuda:0') |
| |
| ########## torch.float32/torch.int32/size=()+(4, 9)+(4, 2) ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([0, 2, 4]), |
| col_indices=tensor([0, 1, 0, 2]), |
| values=tensor([[[[[1.0000e+00, 1.0010e+03], |
| [1.0100e+02, 1.1010e+03], |
| [2.0100e+02, 1.2010e+03], |
| [3.0100e+02, 1.3010e+03]], |
| |
| [[1.1000e+01, 1.0110e+03], |
| [1.1100e+02, 1.1110e+03], |
| [2.1100e+02, 1.2110e+03], |
| [3.1100e+02, 1.3110e+03]], |
| |
| [[2.1000e+01, 1.0210e+03], |
| [1.2100e+02, 1.1210e+03], |
| [2.2100e+02, 1.2210e+03], |
| [3.2100e+02, 1.3210e+03]]], |
| |
| |
| [[[2.0000e+00, 1.0020e+03], |
| [1.0200e+02, 1.1020e+03], |
| [2.0200e+02, 1.2020e+03], |
| [3.0200e+02, 1.3020e+03]], |
| |
| [[1.2000e+01, 1.0120e+03], |
| [1.1200e+02, 1.1120e+03], |
| [2.1200e+02, 1.2120e+03], |
| [3.1200e+02, 1.3120e+03]], |
| |
| [[2.2000e+01, 1.0220e+03], |
| [1.2200e+02, 1.1220e+03], |
| [2.2200e+02, 1.2220e+03], |
| [3.2200e+02, 1.3220e+03]]]], |
| |
| |
| |
| [[[[2.0000e+00, 1.0020e+03], |
| [1.0200e+02, 1.1020e+03], |
| [2.0200e+02, 1.2020e+03], |
| [3.0200e+02, 1.3020e+03]], |
| |
| [[1.2000e+01, 1.0120e+03], |
| [1.1200e+02, 1.1120e+03], |
| [2.1200e+02, 1.2120e+03], |
| [3.1200e+02, 1.3120e+03]], |
| |
| [[2.2000e+01, 1.0220e+03], |
| [1.2200e+02, 1.1220e+03], |
| [2.2200e+02, 1.2220e+03], |
| [3.2200e+02, 1.3220e+03]]], |
| |
| |
| [[[3.0000e+00, 1.0030e+03], |
| [1.0300e+02, 1.1030e+03], |
| [2.0300e+02, 1.2030e+03], |
| [3.0300e+02, 1.3030e+03]], |
| |
| [[1.3000e+01, 1.0130e+03], |
| [1.1300e+02, 1.1130e+03], |
| [2.1300e+02, 1.2130e+03], |
| [3.1300e+02, 1.3130e+03]], |
| |
| [[2.3000e+01, 1.0230e+03], |
| [1.2300e+02, 1.1230e+03], |
| [2.2300e+02, 1.2230e+03], |
| [3.2300e+02, 1.3230e+03]]]], |
| |
| |
| |
| [[[[3.0000e+00, 1.0030e+03], |
| [1.0300e+02, 1.1030e+03], |
| [2.0300e+02, 1.2030e+03], |
| [3.0300e+02, 1.3030e+03]], |
| |
| [[1.3000e+01, 1.0130e+03], |
| [1.1300e+02, 1.1130e+03], |
| [2.1300e+02, 1.2130e+03], |
| [3.1300e+02, 1.3130e+03]], |
| |
| [[2.3000e+01, 1.0230e+03], |
| [1.2300e+02, 1.1230e+03], |
| [2.2300e+02, 1.2230e+03], |
| [3.2300e+02, 1.3230e+03]]], |
| |
| |
| [[[4.0000e+00, 1.0040e+03], |
| [1.0400e+02, 1.1040e+03], |
| [2.0400e+02, 1.2040e+03], |
| [3.0400e+02, 1.3040e+03]], |
| |
| [[1.4000e+01, 1.0140e+03], |
| [1.1400e+02, 1.1140e+03], |
| [2.1400e+02, 1.2140e+03], |
| [3.1400e+02, 1.3140e+03]], |
| |
| [[2.4000e+01, 1.0240e+03], |
| [1.2400e+02, 1.1240e+03], |
| [2.2400e+02, 1.2240e+03], |
| [3.2400e+02, 1.3240e+03]]]], |
| |
| |
| |
| [[[[4.0000e+00, 1.0040e+03], |
| [1.0400e+02, 1.1040e+03], |
| [2.0400e+02, 1.2040e+03], |
| [3.0400e+02, 1.3040e+03]], |
| |
| [[1.4000e+01, 1.0140e+03], |
| [1.1400e+02, 1.1140e+03], |
| [2.1400e+02, 1.2140e+03], |
| [3.1400e+02, 1.3140e+03]], |
| |
| [[2.4000e+01, 1.0240e+03], |
| [1.2400e+02, 1.1240e+03], |
| [2.2400e+02, 1.2240e+03], |
| [3.2400e+02, 1.3240e+03]]], |
| |
| |
| [[[5.0000e+00, 1.0050e+03], |
| [1.0500e+02, 1.1050e+03], |
| [2.0500e+02, 1.2050e+03], |
| [3.0500e+02, 1.3050e+03]], |
| |
| [[1.5000e+01, 1.0150e+03], |
| [1.1500e+02, 1.1150e+03], |
| [2.1500e+02, 1.2150e+03], |
| [3.1500e+02, 1.3150e+03]], |
| |
| [[2.5000e+01, 1.0250e+03], |
| [1.2500e+02, 1.1250e+03], |
| [2.2500e+02, 1.2250e+03], |
| [3.2500e+02, 1.3250e+03]]]]]), device='cuda:0', |
| size=(4, 9, 4, 2), nnz=4, layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([0, 2, 4], device='cuda:0', dtype=torch.int32) |
| # _col_indices |
| tensor([0, 1, 0, 2], device='cuda:0', dtype=torch.int32) |
| # _values |
| tensor([[[[[1.0000e+00, 1.0010e+03], |
| [1.0100e+02, 1.1010e+03], |
| [2.0100e+02, 1.2010e+03], |
| [3.0100e+02, 1.3010e+03]], |
| |
| [[1.1000e+01, 1.0110e+03], |
| [1.1100e+02, 1.1110e+03], |
| [2.1100e+02, 1.2110e+03], |
| [3.1100e+02, 1.3110e+03]], |
| |
| [[2.1000e+01, 1.0210e+03], |
| [1.2100e+02, 1.1210e+03], |
| [2.2100e+02, 1.2210e+03], |
| [3.2100e+02, 1.3210e+03]]], |
| |
| |
| [[[2.0000e+00, 1.0020e+03], |
| [1.0200e+02, 1.1020e+03], |
| [2.0200e+02, 1.2020e+03], |
| [3.0200e+02, 1.3020e+03]], |
| |
| [[1.2000e+01, 1.0120e+03], |
| [1.1200e+02, 1.1120e+03], |
| [2.1200e+02, 1.2120e+03], |
| [3.1200e+02, 1.3120e+03]], |
| |
| [[2.2000e+01, 1.0220e+03], |
| [1.2200e+02, 1.1220e+03], |
| [2.2200e+02, 1.2220e+03], |
| [3.2200e+02, 1.3220e+03]]]], |
| |
| |
| |
| [[[[2.0000e+00, 1.0020e+03], |
| [1.0200e+02, 1.1020e+03], |
| [2.0200e+02, 1.2020e+03], |
| [3.0200e+02, 1.3020e+03]], |
| |
| [[1.2000e+01, 1.0120e+03], |
| [1.1200e+02, 1.1120e+03], |
| [2.1200e+02, 1.2120e+03], |
| [3.1200e+02, 1.3120e+03]], |
| |
| [[2.2000e+01, 1.0220e+03], |
| [1.2200e+02, 1.1220e+03], |
| [2.2200e+02, 1.2220e+03], |
| [3.2200e+02, 1.3220e+03]]], |
| |
| |
| [[[3.0000e+00, 1.0030e+03], |
| [1.0300e+02, 1.1030e+03], |
| [2.0300e+02, 1.2030e+03], |
| [3.0300e+02, 1.3030e+03]], |
| |
| [[1.3000e+01, 1.0130e+03], |
| [1.1300e+02, 1.1130e+03], |
| [2.1300e+02, 1.2130e+03], |
| [3.1300e+02, 1.3130e+03]], |
| |
| [[2.3000e+01, 1.0230e+03], |
| [1.2300e+02, 1.1230e+03], |
| [2.2300e+02, 1.2230e+03], |
| [3.2300e+02, 1.3230e+03]]]], |
| |
| |
| |
| [[[[3.0000e+00, 1.0030e+03], |
| [1.0300e+02, 1.1030e+03], |
| [2.0300e+02, 1.2030e+03], |
| [3.0300e+02, 1.3030e+03]], |
| |
| [[1.3000e+01, 1.0130e+03], |
| [1.1300e+02, 1.1130e+03], |
| [2.1300e+02, 1.2130e+03], |
| [3.1300e+02, 1.3130e+03]], |
| |
| [[2.3000e+01, 1.0230e+03], |
| [1.2300e+02, 1.1230e+03], |
| [2.2300e+02, 1.2230e+03], |
| [3.2300e+02, 1.3230e+03]]], |
| |
| |
| [[[4.0000e+00, 1.0040e+03], |
| [1.0400e+02, 1.1040e+03], |
| [2.0400e+02, 1.2040e+03], |
| [3.0400e+02, 1.3040e+03]], |
| |
| [[1.4000e+01, 1.0140e+03], |
| [1.1400e+02, 1.1140e+03], |
| [2.1400e+02, 1.2140e+03], |
| [3.1400e+02, 1.3140e+03]], |
| |
| [[2.4000e+01, 1.0240e+03], |
| [1.2400e+02, 1.1240e+03], |
| [2.2400e+02, 1.2240e+03], |
| [3.2400e+02, 1.3240e+03]]]], |
| |
| |
| |
| [[[[4.0000e+00, 1.0040e+03], |
| [1.0400e+02, 1.1040e+03], |
| [2.0400e+02, 1.2040e+03], |
| [3.0400e+02, 1.3040e+03]], |
| |
| [[1.4000e+01, 1.0140e+03], |
| [1.1400e+02, 1.1140e+03], |
| [2.1400e+02, 1.2140e+03], |
| [3.1400e+02, 1.3140e+03]], |
| |
| [[2.4000e+01, 1.0240e+03], |
| [1.2400e+02, 1.1240e+03], |
| [2.2400e+02, 1.2240e+03], |
| [3.2400e+02, 1.3240e+03]]], |
| |
| |
| [[[5.0000e+00, 1.0050e+03], |
| [1.0500e+02, 1.1050e+03], |
| [2.0500e+02, 1.2050e+03], |
| [3.0500e+02, 1.3050e+03]], |
| |
| [[1.5000e+01, 1.0150e+03], |
| [1.1500e+02, 1.1150e+03], |
| [2.1500e+02, 1.2150e+03], |
| [3.1500e+02, 1.3150e+03]], |
| |
| [[2.5000e+01, 1.0250e+03], |
| [1.2500e+02, 1.1250e+03], |
| [2.2500e+02, 1.2250e+03], |
| [3.2500e+02, 1.3250e+03]]]]], device='cuda:0') |
| |
| ########## torch.float32/torch.int32/size=(2, 3)+(6, 6)+(2, 1) ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([[[0, 2, 4], |
| [0, 3, 4], |
| [0, 1, 4]], |
| |
| [[0, 1, 4], |
| [0, 2, 4], |
| [0, 3, 4]]]), |
| col_indices=tensor([[[0, 1, 0, 1], |
| [0, 1, 2, 0], |
| [0, 0, 1, 2]], |
| |
| [[1, 0, 1, 2], |
| [0, 2, 0, 1], |
| [0, 1, 2, 1]]]), |
| values=tensor([[[[[[[ 1.], |
| [101.]], |
| |
| [[ 11.], |
| [111.]]], |
| |
| |
| [[[ 2.], |
| [102.]], |
| |
| [[ 12.], |
| [112.]]], |
| |
| |
| [[[ 3.], |
| [103.]], |
| |
| [[ 13.], |
| [113.]]]], |
| |
| |
| |
| [[[[ 2.], |
| [102.]], |
| |
| [[ 12.], |
| [112.]]], |
| |
| |
| [[[ 3.], |
| [103.]], |
| |
| [[ 13.], |
| [113.]]], |
| |
| |
| [[[ 4.], |
| [104.]], |
| |
| [[ 14.], |
| [114.]]]], |
| |
| |
| |
| [[[[ 3.], |
| [103.]], |
| |
| [[ 13.], |
| [113.]]], |
| |
| |
| [[[ 4.], |
| [104.]], |
| |
| [[ 14.], |
| [114.]]], |
| |
| |
| [[[ 5.], |
| [105.]], |
| |
| [[ 15.], |
| [115.]]]], |
| |
| |
| |
| [[[[ 4.], |
| [104.]], |
| |
| [[ 14.], |
| [114.]]], |
| |
| |
| [[[ 5.], |
| [105.]], |
| |
| [[ 15.], |
| [115.]]], |
| |
| |
| [[[ 6.], |
| [106.]], |
| |
| [[ 16.], |
| [116.]]]]], |
| |
| |
| |
| |
| [[[[[ 5.], |
| [105.]], |
| |
| [[ 15.], |
| [115.]]], |
| |
| |
| [[[ 6.], |
| [106.]], |
| |
| [[ 16.], |
| [116.]]], |
| |
| |
| [[[ 7.], |
| [107.]], |
| |
| [[ 17.], |
| [117.]]]], |
| |
| |
| |
| [[[[ 6.], |
| [106.]], |
| |
| [[ 16.], |
| [116.]]], |
| |
| |
| [[[ 7.], |
| [107.]], |
| |
| [[ 17.], |
| [117.]]], |
| |
| |
| [[[ 8.], |
| [108.]], |
| |
| [[ 18.], |
| [118.]]]], |
| |
| |
| |
| [[[[ 7.], |
| [107.]], |
| |
| [[ 17.], |
| [117.]]], |
| |
| |
| [[[ 8.], |
| [108.]], |
| |
| [[ 18.], |
| [118.]]], |
| |
| |
| [[[ 9.], |
| [109.]], |
| |
| [[ 19.], |
| [119.]]]], |
| |
| |
| |
| [[[[ 8.], |
| [108.]], |
| |
| [[ 18.], |
| [118.]]], |
| |
| |
| [[[ 9.], |
| [109.]], |
| |
| [[ 19.], |
| [119.]]], |
| |
| |
| [[[ 10.], |
| [110.]], |
| |
| [[ 20.], |
| [120.]]]]], |
| |
| |
| |
| |
| [[[[[ 9.], |
| [109.]], |
| |
| [[ 19.], |
| [119.]]], |
| |
| |
| [[[ 10.], |
| [110.]], |
| |
| [[ 20.], |
| [120.]]], |
| |
| |
| [[[ 11.], |
| [111.]], |
| |
| [[ 21.], |
| [121.]]]], |
| |
| |
| |
| [[[[ 10.], |
| [110.]], |
| |
| [[ 20.], |
| [120.]]], |
| |
| |
| [[[ 11.], |
| [111.]], |
| |
| [[ 21.], |
| [121.]]], |
| |
| |
| [[[ 12.], |
| [112.]], |
| |
| [[ 22.], |
| [122.]]]], |
| |
| |
| |
| [[[[ 11.], |
| [111.]], |
| |
| [[ 21.], |
| [121.]]], |
| |
| |
| [[[ 12.], |
| [112.]], |
| |
| [[ 22.], |
| [122.]]], |
| |
| |
| [[[ 13.], |
| [113.]], |
| |
| [[ 23.], |
| [123.]]]], |
| |
| |
| |
| [[[[ 12.], |
| [112.]], |
| |
| [[ 22.], |
| [122.]]], |
| |
| |
| [[[ 13.], |
| [113.]], |
| |
| [[ 23.], |
| [123.]]], |
| |
| |
| [[[ 14.], |
| [114.]], |
| |
| [[ 24.], |
| [124.]]]]]], |
| |
| |
| |
| |
| |
| [[[[[[ 13.], |
| [113.]], |
| |
| [[ 23.], |
| [123.]]], |
| |
| |
| [[[ 14.], |
| [114.]], |
| |
| [[ 24.], |
| [124.]]], |
| |
| |
| [[[ 15.], |
| [115.]], |
| |
| [[ 25.], |
| [125.]]]], |
| |
| |
| |
| [[[[ 14.], |
| [114.]], |
| |
| [[ 24.], |
| [124.]]], |
| |
| |
| [[[ 15.], |
| [115.]], |
| |
| [[ 25.], |
| [125.]]], |
| |
| |
| [[[ 16.], |
| [116.]], |
| |
| [[ 26.], |
| [126.]]]], |
| |
| |
| |
| [[[[ 15.], |
| [115.]], |
| |
| [[ 25.], |
| [125.]]], |
| |
| |
| [[[ 16.], |
| [116.]], |
| |
| [[ 26.], |
| [126.]]], |
| |
| |
| [[[ 17.], |
| [117.]], |
| |
| [[ 27.], |
| [127.]]]], |
| |
| |
| |
| [[[[ 16.], |
| [116.]], |
| |
| [[ 26.], |
| [126.]]], |
| |
| |
| [[[ 17.], |
| [117.]], |
| |
| [[ 27.], |
| [127.]]], |
| |
| |
| [[[ 18.], |
| [118.]], |
| |
| [[ 28.], |
| [128.]]]]], |
| |
| |
| |
| |
| [[[[[ 17.], |
| [117.]], |
| |
| [[ 27.], |
| [127.]]], |
| |
| |
| [[[ 18.], |
| [118.]], |
| |
| [[ 28.], |
| [128.]]], |
| |
| |
| [[[ 19.], |
| [119.]], |
| |
| [[ 29.], |
| [129.]]]], |
| |
| |
| |
| [[[[ 18.], |
| [118.]], |
| |
| [[ 28.], |
| [128.]]], |
| |
| |
| [[[ 19.], |
| [119.]], |
| |
| [[ 29.], |
| [129.]]], |
| |
| |
| [[[ 20.], |
| [120.]], |
| |
| [[ 30.], |
| [130.]]]], |
| |
| |
| |
| [[[[ 19.], |
| [119.]], |
| |
| [[ 29.], |
| [129.]]], |
| |
| |
| [[[ 20.], |
| [120.]], |
| |
| [[ 30.], |
| [130.]]], |
| |
| |
| [[[ 21.], |
| [121.]], |
| |
| [[ 31.], |
| [131.]]]], |
| |
| |
| |
| [[[[ 20.], |
| [120.]], |
| |
| [[ 30.], |
| [130.]]], |
| |
| |
| [[[ 21.], |
| [121.]], |
| |
| [[ 31.], |
| [131.]]], |
| |
| |
| [[[ 22.], |
| [122.]], |
| |
| [[ 32.], |
| [132.]]]]], |
| |
| |
| |
| |
| [[[[[ 21.], |
| [121.]], |
| |
| [[ 31.], |
| [131.]]], |
| |
| |
| [[[ 22.], |
| [122.]], |
| |
| [[ 32.], |
| [132.]]], |
| |
| |
| [[[ 23.], |
| [123.]], |
| |
| [[ 33.], |
| [133.]]]], |
| |
| |
| |
| [[[[ 22.], |
| [122.]], |
| |
| [[ 32.], |
| [132.]]], |
| |
| |
| [[[ 23.], |
| [123.]], |
| |
| [[ 33.], |
| [133.]]], |
| |
| |
| [[[ 24.], |
| [124.]], |
| |
| [[ 34.], |
| [134.]]]], |
| |
| |
| |
| [[[[ 23.], |
| [123.]], |
| |
| [[ 33.], |
| [133.]]], |
| |
| |
| [[[ 24.], |
| [124.]], |
| |
| [[ 34.], |
| [134.]]], |
| |
| |
| [[[ 25.], |
| [125.]], |
| |
| [[ 35.], |
| [135.]]]], |
| |
| |
| |
| [[[[ 24.], |
| [124.]], |
| |
| [[ 34.], |
| [134.]]], |
| |
| |
| [[[ 25.], |
| [125.]], |
| |
| [[ 35.], |
| [135.]]], |
| |
| |
| [[[ 26.], |
| [126.]], |
| |
| [[ 36.], |
| [136.]]]]]]]), device='cuda:0', |
| size=(2, 3, 6, 6, 2, 1), nnz=4, layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([[[0, 2, 4], |
| [0, 3, 4], |
| [0, 1, 4]], |
| |
| [[0, 1, 4], |
| [0, 2, 4], |
| [0, 3, 4]]], device='cuda:0', dtype=torch.int32) |
| # _col_indices |
| tensor([[[0, 1, 0, 1], |
| [0, 1, 2, 0], |
| [0, 0, 1, 2]], |
| |
| [[1, 0, 1, 2], |
| [0, 2, 0, 1], |
| [0, 1, 2, 1]]], device='cuda:0', dtype=torch.int32) |
| # _values |
| tensor([[[[[[[ 1.], |
| [101.]], |
| |
| [[ 11.], |
| [111.]]], |
| |
| |
| [[[ 2.], |
| [102.]], |
| |
| [[ 12.], |
| [112.]]], |
| |
| |
| [[[ 3.], |
| [103.]], |
| |
| [[ 13.], |
| [113.]]]], |
| |
| |
| |
| [[[[ 2.], |
| [102.]], |
| |
| [[ 12.], |
| [112.]]], |
| |
| |
| [[[ 3.], |
| [103.]], |
| |
| [[ 13.], |
| [113.]]], |
| |
| |
| [[[ 4.], |
| [104.]], |
| |
| [[ 14.], |
| [114.]]]], |
| |
| |
| |
| [[[[ 3.], |
| [103.]], |
| |
| [[ 13.], |
| [113.]]], |
| |
| |
| [[[ 4.], |
| [104.]], |
| |
| [[ 14.], |
| [114.]]], |
| |
| |
| [[[ 5.], |
| [105.]], |
| |
| [[ 15.], |
| [115.]]]], |
| |
| |
| |
| [[[[ 4.], |
| [104.]], |
| |
| [[ 14.], |
| [114.]]], |
| |
| |
| [[[ 5.], |
| [105.]], |
| |
| [[ 15.], |
| [115.]]], |
| |
| |
| [[[ 6.], |
| [106.]], |
| |
| [[ 16.], |
| [116.]]]]], |
| |
| |
| |
| |
| [[[[[ 5.], |
| [105.]], |
| |
| [[ 15.], |
| [115.]]], |
| |
| |
| [[[ 6.], |
| [106.]], |
| |
| [[ 16.], |
| [116.]]], |
| |
| |
| [[[ 7.], |
| [107.]], |
| |
| [[ 17.], |
| [117.]]]], |
| |
| |
| |
| [[[[ 6.], |
| [106.]], |
| |
| [[ 16.], |
| [116.]]], |
| |
| |
| [[[ 7.], |
| [107.]], |
| |
| [[ 17.], |
| [117.]]], |
| |
| |
| [[[ 8.], |
| [108.]], |
| |
| [[ 18.], |
| [118.]]]], |
| |
| |
| |
| [[[[ 7.], |
| [107.]], |
| |
| [[ 17.], |
| [117.]]], |
| |
| |
| [[[ 8.], |
| [108.]], |
| |
| [[ 18.], |
| [118.]]], |
| |
| |
| [[[ 9.], |
| [109.]], |
| |
| [[ 19.], |
| [119.]]]], |
| |
| |
| |
| [[[[ 8.], |
| [108.]], |
| |
| [[ 18.], |
| [118.]]], |
| |
| |
| [[[ 9.], |
| [109.]], |
| |
| [[ 19.], |
| [119.]]], |
| |
| |
| [[[ 10.], |
| [110.]], |
| |
| [[ 20.], |
| [120.]]]]], |
| |
| |
| |
| |
| [[[[[ 9.], |
| [109.]], |
| |
| [[ 19.], |
| [119.]]], |
| |
| |
| [[[ 10.], |
| [110.]], |
| |
| [[ 20.], |
| [120.]]], |
| |
| |
| [[[ 11.], |
| [111.]], |
| |
| [[ 21.], |
| [121.]]]], |
| |
| |
| |
| [[[[ 10.], |
| [110.]], |
| |
| [[ 20.], |
| [120.]]], |
| |
| |
| [[[ 11.], |
| [111.]], |
| |
| [[ 21.], |
| [121.]]], |
| |
| |
| [[[ 12.], |
| [112.]], |
| |
| [[ 22.], |
| [122.]]]], |
| |
| |
| |
| [[[[ 11.], |
| [111.]], |
| |
| [[ 21.], |
| [121.]]], |
| |
| |
| [[[ 12.], |
| [112.]], |
| |
| [[ 22.], |
| [122.]]], |
| |
| |
| [[[ 13.], |
| [113.]], |
| |
| [[ 23.], |
| [123.]]]], |
| |
| |
| |
| [[[[ 12.], |
| [112.]], |
| |
| [[ 22.], |
| [122.]]], |
| |
| |
| [[[ 13.], |
| [113.]], |
| |
| [[ 23.], |
| [123.]]], |
| |
| |
| [[[ 14.], |
| [114.]], |
| |
| [[ 24.], |
| [124.]]]]]], |
| |
| |
| |
| |
| |
| [[[[[[ 13.], |
| [113.]], |
| |
| [[ 23.], |
| [123.]]], |
| |
| |
| [[[ 14.], |
| [114.]], |
| |
| [[ 24.], |
| [124.]]], |
| |
| |
| [[[ 15.], |
| [115.]], |
| |
| [[ 25.], |
| [125.]]]], |
| |
| |
| |
| [[[[ 14.], |
| [114.]], |
| |
| [[ 24.], |
| [124.]]], |
| |
| |
| [[[ 15.], |
| [115.]], |
| |
| [[ 25.], |
| [125.]]], |
| |
| |
| [[[ 16.], |
| [116.]], |
| |
| [[ 26.], |
| [126.]]]], |
| |
| |
| |
| [[[[ 15.], |
| [115.]], |
| |
| [[ 25.], |
| [125.]]], |
| |
| |
| [[[ 16.], |
| [116.]], |
| |
| [[ 26.], |
| [126.]]], |
| |
| |
| [[[ 17.], |
| [117.]], |
| |
| [[ 27.], |
| [127.]]]], |
| |
| |
| |
| [[[[ 16.], |
| [116.]], |
| |
| [[ 26.], |
| [126.]]], |
| |
| |
| [[[ 17.], |
| [117.]], |
| |
| [[ 27.], |
| [127.]]], |
| |
| |
| [[[ 18.], |
| [118.]], |
| |
| [[ 28.], |
| [128.]]]]], |
| |
| |
| |
| |
| [[[[[ 17.], |
| [117.]], |
| |
| [[ 27.], |
| [127.]]], |
| |
| |
| [[[ 18.], |
| [118.]], |
| |
| [[ 28.], |
| [128.]]], |
| |
| |
| [[[ 19.], |
| [119.]], |
| |
| [[ 29.], |
| [129.]]]], |
| |
| |
| |
| [[[[ 18.], |
| [118.]], |
| |
| [[ 28.], |
| [128.]]], |
| |
| |
| [[[ 19.], |
| [119.]], |
| |
| [[ 29.], |
| [129.]]], |
| |
| |
| [[[ 20.], |
| [120.]], |
| |
| [[ 30.], |
| [130.]]]], |
| |
| |
| |
| [[[[ 19.], |
| [119.]], |
| |
| [[ 29.], |
| [129.]]], |
| |
| |
| [[[ 20.], |
| [120.]], |
| |
| [[ 30.], |
| [130.]]], |
| |
| |
| [[[ 21.], |
| [121.]], |
| |
| [[ 31.], |
| [131.]]]], |
| |
| |
| |
| [[[[ 20.], |
| [120.]], |
| |
| [[ 30.], |
| [130.]]], |
| |
| |
| [[[ 21.], |
| [121.]], |
| |
| [[ 31.], |
| [131.]]], |
| |
| |
| [[[ 22.], |
| [122.]], |
| |
| [[ 32.], |
| [132.]]]]], |
| |
| |
| |
| |
| [[[[[ 21.], |
| [121.]], |
| |
| [[ 31.], |
| [131.]]], |
| |
| |
| [[[ 22.], |
| [122.]], |
| |
| [[ 32.], |
| [132.]]], |
| |
| |
| [[[ 23.], |
| [123.]], |
| |
| [[ 33.], |
| [133.]]]], |
| |
| |
| |
| [[[[ 22.], |
| [122.]], |
| |
| [[ 32.], |
| [132.]]], |
| |
| |
| [[[ 23.], |
| [123.]], |
| |
| [[ 33.], |
| [133.]]], |
| |
| |
| [[[ 24.], |
| [124.]], |
| |
| [[ 34.], |
| [134.]]]], |
| |
| |
| |
| [[[[ 23.], |
| [123.]], |
| |
| [[ 33.], |
| [133.]]], |
| |
| |
| [[[ 24.], |
| [124.]], |
| |
| [[ 34.], |
| [134.]]], |
| |
| |
| [[[ 25.], |
| [125.]], |
| |
| [[ 35.], |
| [135.]]]], |
| |
| |
| |
| [[[[ 24.], |
| [124.]], |
| |
| [[ 34.], |
| [134.]]], |
| |
| |
| [[[ 25.], |
| [125.]], |
| |
| [[ 35.], |
| [135.]]], |
| |
| |
| [[[ 26.], |
| [126.]], |
| |
| [[ 36.], |
| [136.]]]]]]], device='cuda:0') |
| |
| |
| ########## torch.float64/torch.int32/size=()+(6, 6)+(2,) ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([0, 2, 4]), |
| col_indices=tensor([0, 1, 0, 2]), |
| values=tensor([[[[ 1., 101.], |
| [ 11., 111.]], |
| |
| [[ 2., 102.], |
| [ 12., 112.]], |
| |
| [[ 3., 103.], |
| [ 13., 113.]]], |
| |
| |
| [[[ 2., 102.], |
| [ 12., 112.]], |
| |
| [[ 3., 103.], |
| [ 13., 113.]], |
| |
| [[ 4., 104.], |
| [ 14., 114.]]], |
| |
| |
| [[[ 3., 103.], |
| [ 13., 113.]], |
| |
| [[ 4., 104.], |
| [ 14., 114.]], |
| |
| [[ 5., 105.], |
| [ 15., 115.]]], |
| |
| |
| [[[ 4., 104.], |
| [ 14., 114.]], |
| |
| [[ 5., 105.], |
| [ 15., 115.]], |
| |
| [[ 6., 106.], |
| [ 16., 116.]]]]), device='cuda:0', size=(6, 6, 2), |
| nnz=4, dtype=torch.float64, layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([0, 2, 4], device='cuda:0', dtype=torch.int32) |
| # _col_indices |
| tensor([0, 1, 0, 2], device='cuda:0', dtype=torch.int32) |
| # _values |
| tensor([[[[ 1., 101.], |
| [ 11., 111.]], |
| |
| [[ 2., 102.], |
| [ 12., 112.]], |
| |
| [[ 3., 103.], |
| [ 13., 113.]]], |
| |
| |
| [[[ 2., 102.], |
| [ 12., 112.]], |
| |
| [[ 3., 103.], |
| [ 13., 113.]], |
| |
| [[ 4., 104.], |
| [ 14., 114.]]], |
| |
| |
| [[[ 3., 103.], |
| [ 13., 113.]], |
| |
| [[ 4., 104.], |
| [ 14., 114.]], |
| |
| [[ 5., 105.], |
| [ 15., 115.]]], |
| |
| |
| [[[ 4., 104.], |
| [ 14., 114.]], |
| |
| [[ 5., 105.], |
| [ 15., 115.]], |
| |
| [[ 6., 106.], |
| [ 16., 116.]]]], device='cuda:0', dtype=torch.float64) |
| |
| ########## torch.float64/torch.int32/size=()+(4, 9)+(4, 2) ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([0, 2, 4]), |
| col_indices=tensor([0, 1, 0, 2]), |
| values=tensor([[[[[1.0000e+00, 1.0010e+03], |
| [1.0100e+02, 1.1010e+03], |
| [2.0100e+02, 1.2010e+03], |
| [3.0100e+02, 1.3010e+03]], |
| |
| [[1.1000e+01, 1.0110e+03], |
| [1.1100e+02, 1.1110e+03], |
| [2.1100e+02, 1.2110e+03], |
| [3.1100e+02, 1.3110e+03]], |
| |
| [[2.1000e+01, 1.0210e+03], |
| [1.2100e+02, 1.1210e+03], |
| [2.2100e+02, 1.2210e+03], |
| [3.2100e+02, 1.3210e+03]]], |
| |
| |
| [[[2.0000e+00, 1.0020e+03], |
| [1.0200e+02, 1.1020e+03], |
| [2.0200e+02, 1.2020e+03], |
| [3.0200e+02, 1.3020e+03]], |
| |
| [[1.2000e+01, 1.0120e+03], |
| [1.1200e+02, 1.1120e+03], |
| [2.1200e+02, 1.2120e+03], |
| [3.1200e+02, 1.3120e+03]], |
| |
| [[2.2000e+01, 1.0220e+03], |
| [1.2200e+02, 1.1220e+03], |
| [2.2200e+02, 1.2220e+03], |
| [3.2200e+02, 1.3220e+03]]]], |
| |
| |
| |
| [[[[2.0000e+00, 1.0020e+03], |
| [1.0200e+02, 1.1020e+03], |
| [2.0200e+02, 1.2020e+03], |
| [3.0200e+02, 1.3020e+03]], |
| |
| [[1.2000e+01, 1.0120e+03], |
| [1.1200e+02, 1.1120e+03], |
| [2.1200e+02, 1.2120e+03], |
| [3.1200e+02, 1.3120e+03]], |
| |
| [[2.2000e+01, 1.0220e+03], |
| [1.2200e+02, 1.1220e+03], |
| [2.2200e+02, 1.2220e+03], |
| [3.2200e+02, 1.3220e+03]]], |
| |
| |
| [[[3.0000e+00, 1.0030e+03], |
| [1.0300e+02, 1.1030e+03], |
| [2.0300e+02, 1.2030e+03], |
| [3.0300e+02, 1.3030e+03]], |
| |
| [[1.3000e+01, 1.0130e+03], |
| [1.1300e+02, 1.1130e+03], |
| [2.1300e+02, 1.2130e+03], |
| [3.1300e+02, 1.3130e+03]], |
| |
| [[2.3000e+01, 1.0230e+03], |
| [1.2300e+02, 1.1230e+03], |
| [2.2300e+02, 1.2230e+03], |
| [3.2300e+02, 1.3230e+03]]]], |
| |
| |
| |
| [[[[3.0000e+00, 1.0030e+03], |
| [1.0300e+02, 1.1030e+03], |
| [2.0300e+02, 1.2030e+03], |
| [3.0300e+02, 1.3030e+03]], |
| |
| [[1.3000e+01, 1.0130e+03], |
| [1.1300e+02, 1.1130e+03], |
| [2.1300e+02, 1.2130e+03], |
| [3.1300e+02, 1.3130e+03]], |
| |
| [[2.3000e+01, 1.0230e+03], |
| [1.2300e+02, 1.1230e+03], |
| [2.2300e+02, 1.2230e+03], |
| [3.2300e+02, 1.3230e+03]]], |
| |
| |
| [[[4.0000e+00, 1.0040e+03], |
| [1.0400e+02, 1.1040e+03], |
| [2.0400e+02, 1.2040e+03], |
| [3.0400e+02, 1.3040e+03]], |
| |
| [[1.4000e+01, 1.0140e+03], |
| [1.1400e+02, 1.1140e+03], |
| [2.1400e+02, 1.2140e+03], |
| [3.1400e+02, 1.3140e+03]], |
| |
| [[2.4000e+01, 1.0240e+03], |
| [1.2400e+02, 1.1240e+03], |
| [2.2400e+02, 1.2240e+03], |
| [3.2400e+02, 1.3240e+03]]]], |
| |
| |
| |
| [[[[4.0000e+00, 1.0040e+03], |
| [1.0400e+02, 1.1040e+03], |
| [2.0400e+02, 1.2040e+03], |
| [3.0400e+02, 1.3040e+03]], |
| |
| [[1.4000e+01, 1.0140e+03], |
| [1.1400e+02, 1.1140e+03], |
| [2.1400e+02, 1.2140e+03], |
| [3.1400e+02, 1.3140e+03]], |
| |
| [[2.4000e+01, 1.0240e+03], |
| [1.2400e+02, 1.1240e+03], |
| [2.2400e+02, 1.2240e+03], |
| [3.2400e+02, 1.3240e+03]]], |
| |
| |
| [[[5.0000e+00, 1.0050e+03], |
| [1.0500e+02, 1.1050e+03], |
| [2.0500e+02, 1.2050e+03], |
| [3.0500e+02, 1.3050e+03]], |
| |
| [[1.5000e+01, 1.0150e+03], |
| [1.1500e+02, 1.1150e+03], |
| [2.1500e+02, 1.2150e+03], |
| [3.1500e+02, 1.3150e+03]], |
| |
| [[2.5000e+01, 1.0250e+03], |
| [1.2500e+02, 1.1250e+03], |
| [2.2500e+02, 1.2250e+03], |
| [3.2500e+02, 1.3250e+03]]]]]), device='cuda:0', |
| size=(4, 9, 4, 2), nnz=4, dtype=torch.float64, layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([0, 2, 4], device='cuda:0', dtype=torch.int32) |
| # _col_indices |
| tensor([0, 1, 0, 2], device='cuda:0', dtype=torch.int32) |
| # _values |
| tensor([[[[[1.0000e+00, 1.0010e+03], |
| [1.0100e+02, 1.1010e+03], |
| [2.0100e+02, 1.2010e+03], |
| [3.0100e+02, 1.3010e+03]], |
| |
| [[1.1000e+01, 1.0110e+03], |
| [1.1100e+02, 1.1110e+03], |
| [2.1100e+02, 1.2110e+03], |
| [3.1100e+02, 1.3110e+03]], |
| |
| [[2.1000e+01, 1.0210e+03], |
| [1.2100e+02, 1.1210e+03], |
| [2.2100e+02, 1.2210e+03], |
| [3.2100e+02, 1.3210e+03]]], |
| |
| |
| [[[2.0000e+00, 1.0020e+03], |
| [1.0200e+02, 1.1020e+03], |
| [2.0200e+02, 1.2020e+03], |
| [3.0200e+02, 1.3020e+03]], |
| |
| [[1.2000e+01, 1.0120e+03], |
| [1.1200e+02, 1.1120e+03], |
| [2.1200e+02, 1.2120e+03], |
| [3.1200e+02, 1.3120e+03]], |
| |
| [[2.2000e+01, 1.0220e+03], |
| [1.2200e+02, 1.1220e+03], |
| [2.2200e+02, 1.2220e+03], |
| [3.2200e+02, 1.3220e+03]]]], |
| |
| |
| |
| [[[[2.0000e+00, 1.0020e+03], |
| [1.0200e+02, 1.1020e+03], |
| [2.0200e+02, 1.2020e+03], |
| [3.0200e+02, 1.3020e+03]], |
| |
| [[1.2000e+01, 1.0120e+03], |
| [1.1200e+02, 1.1120e+03], |
| [2.1200e+02, 1.2120e+03], |
| [3.1200e+02, 1.3120e+03]], |
| |
| [[2.2000e+01, 1.0220e+03], |
| [1.2200e+02, 1.1220e+03], |
| [2.2200e+02, 1.2220e+03], |
| [3.2200e+02, 1.3220e+03]]], |
| |
| |
| [[[3.0000e+00, 1.0030e+03], |
| [1.0300e+02, 1.1030e+03], |
| [2.0300e+02, 1.2030e+03], |
| [3.0300e+02, 1.3030e+03]], |
| |
| [[1.3000e+01, 1.0130e+03], |
| [1.1300e+02, 1.1130e+03], |
| [2.1300e+02, 1.2130e+03], |
| [3.1300e+02, 1.3130e+03]], |
| |
| [[2.3000e+01, 1.0230e+03], |
| [1.2300e+02, 1.1230e+03], |
| [2.2300e+02, 1.2230e+03], |
| [3.2300e+02, 1.3230e+03]]]], |
| |
| |
| |
| [[[[3.0000e+00, 1.0030e+03], |
| [1.0300e+02, 1.1030e+03], |
| [2.0300e+02, 1.2030e+03], |
| [3.0300e+02, 1.3030e+03]], |
| |
| [[1.3000e+01, 1.0130e+03], |
| [1.1300e+02, 1.1130e+03], |
| [2.1300e+02, 1.2130e+03], |
| [3.1300e+02, 1.3130e+03]], |
| |
| [[2.3000e+01, 1.0230e+03], |
| [1.2300e+02, 1.1230e+03], |
| [2.2300e+02, 1.2230e+03], |
| [3.2300e+02, 1.3230e+03]]], |
| |
| |
| [[[4.0000e+00, 1.0040e+03], |
| [1.0400e+02, 1.1040e+03], |
| [2.0400e+02, 1.2040e+03], |
| [3.0400e+02, 1.3040e+03]], |
| |
| [[1.4000e+01, 1.0140e+03], |
| [1.1400e+02, 1.1140e+03], |
| [2.1400e+02, 1.2140e+03], |
| [3.1400e+02, 1.3140e+03]], |
| |
| [[2.4000e+01, 1.0240e+03], |
| [1.2400e+02, 1.1240e+03], |
| [2.2400e+02, 1.2240e+03], |
| [3.2400e+02, 1.3240e+03]]]], |
| |
| |
| |
| [[[[4.0000e+00, 1.0040e+03], |
| [1.0400e+02, 1.1040e+03], |
| [2.0400e+02, 1.2040e+03], |
| [3.0400e+02, 1.3040e+03]], |
| |
| [[1.4000e+01, 1.0140e+03], |
| [1.1400e+02, 1.1140e+03], |
| [2.1400e+02, 1.2140e+03], |
| [3.1400e+02, 1.3140e+03]], |
| |
| [[2.4000e+01, 1.0240e+03], |
| [1.2400e+02, 1.1240e+03], |
| [2.2400e+02, 1.2240e+03], |
| [3.2400e+02, 1.3240e+03]]], |
| |
| |
| [[[5.0000e+00, 1.0050e+03], |
| [1.0500e+02, 1.1050e+03], |
| [2.0500e+02, 1.2050e+03], |
| [3.0500e+02, 1.3050e+03]], |
| |
| [[1.5000e+01, 1.0150e+03], |
| [1.1500e+02, 1.1150e+03], |
| [2.1500e+02, 1.2150e+03], |
| [3.1500e+02, 1.3150e+03]], |
| |
| [[2.5000e+01, 1.0250e+03], |
| [1.2500e+02, 1.1250e+03], |
| [2.2500e+02, 1.2250e+03], |
| [3.2500e+02, 1.3250e+03]]]]], device='cuda:0', dtype=torch.float64) |
| |
| ########## torch.float64/torch.int32/size=(2, 3)+(6, 6)+(2, 1) ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([[[0, 2, 4], |
| [0, 3, 4], |
| [0, 1, 4]], |
| |
| [[0, 1, 4], |
| [0, 2, 4], |
| [0, 3, 4]]]), |
| col_indices=tensor([[[0, 1, 0, 1], |
| [0, 1, 2, 0], |
| [0, 0, 1, 2]], |
| |
| [[1, 0, 1, 2], |
| [0, 2, 0, 1], |
| [0, 1, 2, 1]]]), |
| values=tensor([[[[[[[ 1.], |
| [101.]], |
| |
| [[ 11.], |
| [111.]]], |
| |
| |
| [[[ 2.], |
| [102.]], |
| |
| [[ 12.], |
| [112.]]], |
| |
| |
| [[[ 3.], |
| [103.]], |
| |
| [[ 13.], |
| [113.]]]], |
| |
| |
| |
| [[[[ 2.], |
| [102.]], |
| |
| [[ 12.], |
| [112.]]], |
| |
| |
| [[[ 3.], |
| [103.]], |
| |
| [[ 13.], |
| [113.]]], |
| |
| |
| [[[ 4.], |
| [104.]], |
| |
| [[ 14.], |
| [114.]]]], |
| |
| |
| |
| [[[[ 3.], |
| [103.]], |
| |
| [[ 13.], |
| [113.]]], |
| |
| |
| [[[ 4.], |
| [104.]], |
| |
| [[ 14.], |
| [114.]]], |
| |
| |
| [[[ 5.], |
| [105.]], |
| |
| [[ 15.], |
| [115.]]]], |
| |
| |
| |
| [[[[ 4.], |
| [104.]], |
| |
| [[ 14.], |
| [114.]]], |
| |
| |
| [[[ 5.], |
| [105.]], |
| |
| [[ 15.], |
| [115.]]], |
| |
| |
| [[[ 6.], |
| [106.]], |
| |
| [[ 16.], |
| [116.]]]]], |
| |
| |
| |
| |
| [[[[[ 5.], |
| [105.]], |
| |
| [[ 15.], |
| [115.]]], |
| |
| |
| [[[ 6.], |
| [106.]], |
| |
| [[ 16.], |
| [116.]]], |
| |
| |
| [[[ 7.], |
| [107.]], |
| |
| [[ 17.], |
| [117.]]]], |
| |
| |
| |
| [[[[ 6.], |
| [106.]], |
| |
| [[ 16.], |
| [116.]]], |
| |
| |
| [[[ 7.], |
| [107.]], |
| |
| [[ 17.], |
| [117.]]], |
| |
| |
| [[[ 8.], |
| [108.]], |
| |
| [[ 18.], |
| [118.]]]], |
| |
| |
| |
| [[[[ 7.], |
| [107.]], |
| |
| [[ 17.], |
| [117.]]], |
| |
| |
| [[[ 8.], |
| [108.]], |
| |
| [[ 18.], |
| [118.]]], |
| |
| |
| [[[ 9.], |
| [109.]], |
| |
| [[ 19.], |
| [119.]]]], |
| |
| |
| |
| [[[[ 8.], |
| [108.]], |
| |
| [[ 18.], |
| [118.]]], |
| |
| |
| [[[ 9.], |
| [109.]], |
| |
| [[ 19.], |
| [119.]]], |
| |
| |
| [[[ 10.], |
| [110.]], |
| |
| [[ 20.], |
| [120.]]]]], |
| |
| |
| |
| |
| [[[[[ 9.], |
| [109.]], |
| |
| [[ 19.], |
| [119.]]], |
| |
| |
| [[[ 10.], |
| [110.]], |
| |
| [[ 20.], |
| [120.]]], |
| |
| |
| [[[ 11.], |
| [111.]], |
| |
| [[ 21.], |
| [121.]]]], |
| |
| |
| |
| [[[[ 10.], |
| [110.]], |
| |
| [[ 20.], |
| [120.]]], |
| |
| |
| [[[ 11.], |
| [111.]], |
| |
| [[ 21.], |
| [121.]]], |
| |
| |
| [[[ 12.], |
| [112.]], |
| |
| [[ 22.], |
| [122.]]]], |
| |
| |
| |
| [[[[ 11.], |
| [111.]], |
| |
| [[ 21.], |
| [121.]]], |
| |
| |
| [[[ 12.], |
| [112.]], |
| |
| [[ 22.], |
| [122.]]], |
| |
| |
| [[[ 13.], |
| [113.]], |
| |
| [[ 23.], |
| [123.]]]], |
| |
| |
| |
| [[[[ 12.], |
| [112.]], |
| |
| [[ 22.], |
| [122.]]], |
| |
| |
| [[[ 13.], |
| [113.]], |
| |
| [[ 23.], |
| [123.]]], |
| |
| |
| [[[ 14.], |
| [114.]], |
| |
| [[ 24.], |
| [124.]]]]]], |
| |
| |
| |
| |
| |
| [[[[[[ 13.], |
| [113.]], |
| |
| [[ 23.], |
| [123.]]], |
| |
| |
| [[[ 14.], |
| [114.]], |
| |
| [[ 24.], |
| [124.]]], |
| |
| |
| [[[ 15.], |
| [115.]], |
| |
| [[ 25.], |
| [125.]]]], |
| |
| |
| |
| [[[[ 14.], |
| [114.]], |
| |
| [[ 24.], |
| [124.]]], |
| |
| |
| [[[ 15.], |
| [115.]], |
| |
| [[ 25.], |
| [125.]]], |
| |
| |
| [[[ 16.], |
| [116.]], |
| |
| [[ 26.], |
| [126.]]]], |
| |
| |
| |
| [[[[ 15.], |
| [115.]], |
| |
| [[ 25.], |
| [125.]]], |
| |
| |
| [[[ 16.], |
| [116.]], |
| |
| [[ 26.], |
| [126.]]], |
| |
| |
| [[[ 17.], |
| [117.]], |
| |
| [[ 27.], |
| [127.]]]], |
| |
| |
| |
| [[[[ 16.], |
| [116.]], |
| |
| [[ 26.], |
| [126.]]], |
| |
| |
| [[[ 17.], |
| [117.]], |
| |
| [[ 27.], |
| [127.]]], |
| |
| |
| [[[ 18.], |
| [118.]], |
| |
| [[ 28.], |
| [128.]]]]], |
| |
| |
| |
| |
| [[[[[ 17.], |
| [117.]], |
| |
| [[ 27.], |
| [127.]]], |
| |
| |
| [[[ 18.], |
| [118.]], |
| |
| [[ 28.], |
| [128.]]], |
| |
| |
| [[[ 19.], |
| [119.]], |
| |
| [[ 29.], |
| [129.]]]], |
| |
| |
| |
| [[[[ 18.], |
| [118.]], |
| |
| [[ 28.], |
| [128.]]], |
| |
| |
| [[[ 19.], |
| [119.]], |
| |
| [[ 29.], |
| [129.]]], |
| |
| |
| [[[ 20.], |
| [120.]], |
| |
| [[ 30.], |
| [130.]]]], |
| |
| |
| |
| [[[[ 19.], |
| [119.]], |
| |
| [[ 29.], |
| [129.]]], |
| |
| |
| [[[ 20.], |
| [120.]], |
| |
| [[ 30.], |
| [130.]]], |
| |
| |
| [[[ 21.], |
| [121.]], |
| |
| [[ 31.], |
| [131.]]]], |
| |
| |
| |
| [[[[ 20.], |
| [120.]], |
| |
| [[ 30.], |
| [130.]]], |
| |
| |
| [[[ 21.], |
| [121.]], |
| |
| [[ 31.], |
| [131.]]], |
| |
| |
| [[[ 22.], |
| [122.]], |
| |
| [[ 32.], |
| [132.]]]]], |
| |
| |
| |
| |
| [[[[[ 21.], |
| [121.]], |
| |
| [[ 31.], |
| [131.]]], |
| |
| |
| [[[ 22.], |
| [122.]], |
| |
| [[ 32.], |
| [132.]]], |
| |
| |
| [[[ 23.], |
| [123.]], |
| |
| [[ 33.], |
| [133.]]]], |
| |
| |
| |
| [[[[ 22.], |
| [122.]], |
| |
| [[ 32.], |
| [132.]]], |
| |
| |
| [[[ 23.], |
| [123.]], |
| |
| [[ 33.], |
| [133.]]], |
| |
| |
| [[[ 24.], |
| [124.]], |
| |
| [[ 34.], |
| [134.]]]], |
| |
| |
| |
| [[[[ 23.], |
| [123.]], |
| |
| [[ 33.], |
| [133.]]], |
| |
| |
| [[[ 24.], |
| [124.]], |
| |
| [[ 34.], |
| [134.]]], |
| |
| |
| [[[ 25.], |
| [125.]], |
| |
| [[ 35.], |
| [135.]]]], |
| |
| |
| |
| [[[[ 24.], |
| [124.]], |
| |
| [[ 34.], |
| [134.]]], |
| |
| |
| [[[ 25.], |
| [125.]], |
| |
| [[ 35.], |
| [135.]]], |
| |
| |
| [[[ 26.], |
| [126.]], |
| |
| [[ 36.], |
| [136.]]]]]]]), device='cuda:0', |
| size=(2, 3, 6, 6, 2, 1), nnz=4, dtype=torch.float64, |
| layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([[[0, 2, 4], |
| [0, 3, 4], |
| [0, 1, 4]], |
| |
| [[0, 1, 4], |
| [0, 2, 4], |
| [0, 3, 4]]], device='cuda:0', dtype=torch.int32) |
| # _col_indices |
| tensor([[[0, 1, 0, 1], |
| [0, 1, 2, 0], |
| [0, 0, 1, 2]], |
| |
| [[1, 0, 1, 2], |
| [0, 2, 0, 1], |
| [0, 1, 2, 1]]], device='cuda:0', dtype=torch.int32) |
| # _values |
| tensor([[[[[[[ 1.], |
| [101.]], |
| |
| [[ 11.], |
| [111.]]], |
| |
| |
| [[[ 2.], |
| [102.]], |
| |
| [[ 12.], |
| [112.]]], |
| |
| |
| [[[ 3.], |
| [103.]], |
| |
| [[ 13.], |
| [113.]]]], |
| |
| |
| |
| [[[[ 2.], |
| [102.]], |
| |
| [[ 12.], |
| [112.]]], |
| |
| |
| [[[ 3.], |
| [103.]], |
| |
| [[ 13.], |
| [113.]]], |
| |
| |
| [[[ 4.], |
| [104.]], |
| |
| [[ 14.], |
| [114.]]]], |
| |
| |
| |
| [[[[ 3.], |
| [103.]], |
| |
| [[ 13.], |
| [113.]]], |
| |
| |
| [[[ 4.], |
| [104.]], |
| |
| [[ 14.], |
| [114.]]], |
| |
| |
| [[[ 5.], |
| [105.]], |
| |
| [[ 15.], |
| [115.]]]], |
| |
| |
| |
| [[[[ 4.], |
| [104.]], |
| |
| [[ 14.], |
| [114.]]], |
| |
| |
| [[[ 5.], |
| [105.]], |
| |
| [[ 15.], |
| [115.]]], |
| |
| |
| [[[ 6.], |
| [106.]], |
| |
| [[ 16.], |
| [116.]]]]], |
| |
| |
| |
| |
| [[[[[ 5.], |
| [105.]], |
| |
| [[ 15.], |
| [115.]]], |
| |
| |
| [[[ 6.], |
| [106.]], |
| |
| [[ 16.], |
| [116.]]], |
| |
| |
| [[[ 7.], |
| [107.]], |
| |
| [[ 17.], |
| [117.]]]], |
| |
| |
| |
| [[[[ 6.], |
| [106.]], |
| |
| [[ 16.], |
| [116.]]], |
| |
| |
| [[[ 7.], |
| [107.]], |
| |
| [[ 17.], |
| [117.]]], |
| |
| |
| [[[ 8.], |
| [108.]], |
| |
| [[ 18.], |
| [118.]]]], |
| |
| |
| |
| [[[[ 7.], |
| [107.]], |
| |
| [[ 17.], |
| [117.]]], |
| |
| |
| [[[ 8.], |
| [108.]], |
| |
| [[ 18.], |
| [118.]]], |
| |
| |
| [[[ 9.], |
| [109.]], |
| |
| [[ 19.], |
| [119.]]]], |
| |
| |
| |
| [[[[ 8.], |
| [108.]], |
| |
| [[ 18.], |
| [118.]]], |
| |
| |
| [[[ 9.], |
| [109.]], |
| |
| [[ 19.], |
| [119.]]], |
| |
| |
| [[[ 10.], |
| [110.]], |
| |
| [[ 20.], |
| [120.]]]]], |
| |
| |
| |
| |
| [[[[[ 9.], |
| [109.]], |
| |
| [[ 19.], |
| [119.]]], |
| |
| |
| [[[ 10.], |
| [110.]], |
| |
| [[ 20.], |
| [120.]]], |
| |
| |
| [[[ 11.], |
| [111.]], |
| |
| [[ 21.], |
| [121.]]]], |
| |
| |
| |
| [[[[ 10.], |
| [110.]], |
| |
| [[ 20.], |
| [120.]]], |
| |
| |
| [[[ 11.], |
| [111.]], |
| |
| [[ 21.], |
| [121.]]], |
| |
| |
| [[[ 12.], |
| [112.]], |
| |
| [[ 22.], |
| [122.]]]], |
| |
| |
| |
| [[[[ 11.], |
| [111.]], |
| |
| [[ 21.], |
| [121.]]], |
| |
| |
| [[[ 12.], |
| [112.]], |
| |
| [[ 22.], |
| [122.]]], |
| |
| |
| [[[ 13.], |
| [113.]], |
| |
| [[ 23.], |
| [123.]]]], |
| |
| |
| |
| [[[[ 12.], |
| [112.]], |
| |
| [[ 22.], |
| [122.]]], |
| |
| |
| [[[ 13.], |
| [113.]], |
| |
| [[ 23.], |
| [123.]]], |
| |
| |
| [[[ 14.], |
| [114.]], |
| |
| [[ 24.], |
| [124.]]]]]], |
| |
| |
| |
| |
| |
| [[[[[[ 13.], |
| [113.]], |
| |
| [[ 23.], |
| [123.]]], |
| |
| |
| [[[ 14.], |
| [114.]], |
| |
| [[ 24.], |
| [124.]]], |
| |
| |
| [[[ 15.], |
| [115.]], |
| |
| [[ 25.], |
| [125.]]]], |
| |
| |
| |
| [[[[ 14.], |
| [114.]], |
| |
| [[ 24.], |
| [124.]]], |
| |
| |
| [[[ 15.], |
| [115.]], |
| |
| [[ 25.], |
| [125.]]], |
| |
| |
| [[[ 16.], |
| [116.]], |
| |
| [[ 26.], |
| [126.]]]], |
| |
| |
| |
| [[[[ 15.], |
| [115.]], |
| |
| [[ 25.], |
| [125.]]], |
| |
| |
| [[[ 16.], |
| [116.]], |
| |
| [[ 26.], |
| [126.]]], |
| |
| |
| [[[ 17.], |
| [117.]], |
| |
| [[ 27.], |
| [127.]]]], |
| |
| |
| |
| [[[[ 16.], |
| [116.]], |
| |
| [[ 26.], |
| [126.]]], |
| |
| |
| [[[ 17.], |
| [117.]], |
| |
| [[ 27.], |
| [127.]]], |
| |
| |
| [[[ 18.], |
| [118.]], |
| |
| [[ 28.], |
| [128.]]]]], |
| |
| |
| |
| |
| [[[[[ 17.], |
| [117.]], |
| |
| [[ 27.], |
| [127.]]], |
| |
| |
| [[[ 18.], |
| [118.]], |
| |
| [[ 28.], |
| [128.]]], |
| |
| |
| [[[ 19.], |
| [119.]], |
| |
| [[ 29.], |
| [129.]]]], |
| |
| |
| |
| [[[[ 18.], |
| [118.]], |
| |
| [[ 28.], |
| [128.]]], |
| |
| |
| [[[ 19.], |
| [119.]], |
| |
| [[ 29.], |
| [129.]]], |
| |
| |
| [[[ 20.], |
| [120.]], |
| |
| [[ 30.], |
| [130.]]]], |
| |
| |
| |
| [[[[ 19.], |
| [119.]], |
| |
| [[ 29.], |
| [129.]]], |
| |
| |
| [[[ 20.], |
| [120.]], |
| |
| [[ 30.], |
| [130.]]], |
| |
| |
| [[[ 21.], |
| [121.]], |
| |
| [[ 31.], |
| [131.]]]], |
| |
| |
| |
| [[[[ 20.], |
| [120.]], |
| |
| [[ 30.], |
| [130.]]], |
| |
| |
| [[[ 21.], |
| [121.]], |
| |
| [[ 31.], |
| [131.]]], |
| |
| |
| [[[ 22.], |
| [122.]], |
| |
| [[ 32.], |
| [132.]]]]], |
| |
| |
| |
| |
| [[[[[ 21.], |
| [121.]], |
| |
| [[ 31.], |
| [131.]]], |
| |
| |
| [[[ 22.], |
| [122.]], |
| |
| [[ 32.], |
| [132.]]], |
| |
| |
| [[[ 23.], |
| [123.]], |
| |
| [[ 33.], |
| [133.]]]], |
| |
| |
| |
| [[[[ 22.], |
| [122.]], |
| |
| [[ 32.], |
| [132.]]], |
| |
| |
| [[[ 23.], |
| [123.]], |
| |
| [[ 33.], |
| [133.]]], |
| |
| |
| [[[ 24.], |
| [124.]], |
| |
| [[ 34.], |
| [134.]]]], |
| |
| |
| |
| [[[[ 23.], |
| [123.]], |
| |
| [[ 33.], |
| [133.]]], |
| |
| |
| [[[ 24.], |
| [124.]], |
| |
| [[ 34.], |
| [134.]]], |
| |
| |
| [[[ 25.], |
| [125.]], |
| |
| [[ 35.], |
| [135.]]]], |
| |
| |
| |
| [[[[ 24.], |
| [124.]], |
| |
| [[ 34.], |
| [134.]]], |
| |
| |
| [[[ 25.], |
| [125.]], |
| |
| [[ 35.], |
| [135.]]], |
| |
| |
| [[[ 26.], |
| [126.]], |
| |
| [[ 36.], |
| [136.]]]]]]], device='cuda:0', dtype=torch.float64) |
| |
| |
| ########## torch.float32/torch.int64/size=()+(6, 6)+(2,) ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([0, 2, 4]), |
| col_indices=tensor([0, 1, 0, 2]), |
| values=tensor([[[[ 1., 101.], |
| [ 11., 111.]], |
| |
| [[ 2., 102.], |
| [ 12., 112.]], |
| |
| [[ 3., 103.], |
| [ 13., 113.]]], |
| |
| |
| [[[ 2., 102.], |
| [ 12., 112.]], |
| |
| [[ 3., 103.], |
| [ 13., 113.]], |
| |
| [[ 4., 104.], |
| [ 14., 114.]]], |
| |
| |
| [[[ 3., 103.], |
| [ 13., 113.]], |
| |
| [[ 4., 104.], |
| [ 14., 114.]], |
| |
| [[ 5., 105.], |
| [ 15., 115.]]], |
| |
| |
| [[[ 4., 104.], |
| [ 14., 114.]], |
| |
| [[ 5., 105.], |
| [ 15., 115.]], |
| |
| [[ 6., 106.], |
| [ 16., 116.]]]]), device='cuda:0', size=(6, 6, 2), |
| nnz=4, layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([0, 2, 4], device='cuda:0') |
| # _col_indices |
| tensor([0, 1, 0, 2], device='cuda:0') |
| # _values |
| tensor([[[[ 1., 101.], |
| [ 11., 111.]], |
| |
| [[ 2., 102.], |
| [ 12., 112.]], |
| |
| [[ 3., 103.], |
| [ 13., 113.]]], |
| |
| |
| [[[ 2., 102.], |
| [ 12., 112.]], |
| |
| [[ 3., 103.], |
| [ 13., 113.]], |
| |
| [[ 4., 104.], |
| [ 14., 114.]]], |
| |
| |
| [[[ 3., 103.], |
| [ 13., 113.]], |
| |
| [[ 4., 104.], |
| [ 14., 114.]], |
| |
| [[ 5., 105.], |
| [ 15., 115.]]], |
| |
| |
| [[[ 4., 104.], |
| [ 14., 114.]], |
| |
| [[ 5., 105.], |
| [ 15., 115.]], |
| |
| [[ 6., 106.], |
| [ 16., 116.]]]], device='cuda:0') |
| |
| ########## torch.float32/torch.int64/size=()+(4, 9)+(4, 2) ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([0, 2, 4]), |
| col_indices=tensor([0, 1, 0, 2]), |
| values=tensor([[[[[1.0000e+00, 1.0010e+03], |
| [1.0100e+02, 1.1010e+03], |
| [2.0100e+02, 1.2010e+03], |
| [3.0100e+02, 1.3010e+03]], |
| |
| [[1.1000e+01, 1.0110e+03], |
| [1.1100e+02, 1.1110e+03], |
| [2.1100e+02, 1.2110e+03], |
| [3.1100e+02, 1.3110e+03]], |
| |
| [[2.1000e+01, 1.0210e+03], |
| [1.2100e+02, 1.1210e+03], |
| [2.2100e+02, 1.2210e+03], |
| [3.2100e+02, 1.3210e+03]]], |
| |
| |
| [[[2.0000e+00, 1.0020e+03], |
| [1.0200e+02, 1.1020e+03], |
| [2.0200e+02, 1.2020e+03], |
| [3.0200e+02, 1.3020e+03]], |
| |
| [[1.2000e+01, 1.0120e+03], |
| [1.1200e+02, 1.1120e+03], |
| [2.1200e+02, 1.2120e+03], |
| [3.1200e+02, 1.3120e+03]], |
| |
| [[2.2000e+01, 1.0220e+03], |
| [1.2200e+02, 1.1220e+03], |
| [2.2200e+02, 1.2220e+03], |
| [3.2200e+02, 1.3220e+03]]]], |
| |
| |
| |
| [[[[2.0000e+00, 1.0020e+03], |
| [1.0200e+02, 1.1020e+03], |
| [2.0200e+02, 1.2020e+03], |
| [3.0200e+02, 1.3020e+03]], |
| |
| [[1.2000e+01, 1.0120e+03], |
| [1.1200e+02, 1.1120e+03], |
| [2.1200e+02, 1.2120e+03], |
| [3.1200e+02, 1.3120e+03]], |
| |
| [[2.2000e+01, 1.0220e+03], |
| [1.2200e+02, 1.1220e+03], |
| [2.2200e+02, 1.2220e+03], |
| [3.2200e+02, 1.3220e+03]]], |
| |
| |
| [[[3.0000e+00, 1.0030e+03], |
| [1.0300e+02, 1.1030e+03], |
| [2.0300e+02, 1.2030e+03], |
| [3.0300e+02, 1.3030e+03]], |
| |
| [[1.3000e+01, 1.0130e+03], |
| [1.1300e+02, 1.1130e+03], |
| [2.1300e+02, 1.2130e+03], |
| [3.1300e+02, 1.3130e+03]], |
| |
| [[2.3000e+01, 1.0230e+03], |
| [1.2300e+02, 1.1230e+03], |
| [2.2300e+02, 1.2230e+03], |
| [3.2300e+02, 1.3230e+03]]]], |
| |
| |
| |
| [[[[3.0000e+00, 1.0030e+03], |
| [1.0300e+02, 1.1030e+03], |
| [2.0300e+02, 1.2030e+03], |
| [3.0300e+02, 1.3030e+03]], |
| |
| [[1.3000e+01, 1.0130e+03], |
| [1.1300e+02, 1.1130e+03], |
| [2.1300e+02, 1.2130e+03], |
| [3.1300e+02, 1.3130e+03]], |
| |
| [[2.3000e+01, 1.0230e+03], |
| [1.2300e+02, 1.1230e+03], |
| [2.2300e+02, 1.2230e+03], |
| [3.2300e+02, 1.3230e+03]]], |
| |
| |
| [[[4.0000e+00, 1.0040e+03], |
| [1.0400e+02, 1.1040e+03], |
| [2.0400e+02, 1.2040e+03], |
| [3.0400e+02, 1.3040e+03]], |
| |
| [[1.4000e+01, 1.0140e+03], |
| [1.1400e+02, 1.1140e+03], |
| [2.1400e+02, 1.2140e+03], |
| [3.1400e+02, 1.3140e+03]], |
| |
| [[2.4000e+01, 1.0240e+03], |
| [1.2400e+02, 1.1240e+03], |
| [2.2400e+02, 1.2240e+03], |
| [3.2400e+02, 1.3240e+03]]]], |
| |
| |
| |
| [[[[4.0000e+00, 1.0040e+03], |
| [1.0400e+02, 1.1040e+03], |
| [2.0400e+02, 1.2040e+03], |
| [3.0400e+02, 1.3040e+03]], |
| |
| [[1.4000e+01, 1.0140e+03], |
| [1.1400e+02, 1.1140e+03], |
| [2.1400e+02, 1.2140e+03], |
| [3.1400e+02, 1.3140e+03]], |
| |
| [[2.4000e+01, 1.0240e+03], |
| [1.2400e+02, 1.1240e+03], |
| [2.2400e+02, 1.2240e+03], |
| [3.2400e+02, 1.3240e+03]]], |
| |
| |
| [[[5.0000e+00, 1.0050e+03], |
| [1.0500e+02, 1.1050e+03], |
| [2.0500e+02, 1.2050e+03], |
| [3.0500e+02, 1.3050e+03]], |
| |
| [[1.5000e+01, 1.0150e+03], |
| [1.1500e+02, 1.1150e+03], |
| [2.1500e+02, 1.2150e+03], |
| [3.1500e+02, 1.3150e+03]], |
| |
| [[2.5000e+01, 1.0250e+03], |
| [1.2500e+02, 1.1250e+03], |
| [2.2500e+02, 1.2250e+03], |
| [3.2500e+02, 1.3250e+03]]]]]), device='cuda:0', |
| size=(4, 9, 4, 2), nnz=4, layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([0, 2, 4], device='cuda:0') |
| # _col_indices |
| tensor([0, 1, 0, 2], device='cuda:0') |
| # _values |
| tensor([[[[[1.0000e+00, 1.0010e+03], |
| [1.0100e+02, 1.1010e+03], |
| [2.0100e+02, 1.2010e+03], |
| [3.0100e+02, 1.3010e+03]], |
| |
| [[1.1000e+01, 1.0110e+03], |
| [1.1100e+02, 1.1110e+03], |
| [2.1100e+02, 1.2110e+03], |
| [3.1100e+02, 1.3110e+03]], |
| |
| [[2.1000e+01, 1.0210e+03], |
| [1.2100e+02, 1.1210e+03], |
| [2.2100e+02, 1.2210e+03], |
| [3.2100e+02, 1.3210e+03]]], |
| |
| |
| [[[2.0000e+00, 1.0020e+03], |
| [1.0200e+02, 1.1020e+03], |
| [2.0200e+02, 1.2020e+03], |
| [3.0200e+02, 1.3020e+03]], |
| |
| [[1.2000e+01, 1.0120e+03], |
| [1.1200e+02, 1.1120e+03], |
| [2.1200e+02, 1.2120e+03], |
| [3.1200e+02, 1.3120e+03]], |
| |
| [[2.2000e+01, 1.0220e+03], |
| [1.2200e+02, 1.1220e+03], |
| [2.2200e+02, 1.2220e+03], |
| [3.2200e+02, 1.3220e+03]]]], |
| |
| |
| |
| [[[[2.0000e+00, 1.0020e+03], |
| [1.0200e+02, 1.1020e+03], |
| [2.0200e+02, 1.2020e+03], |
| [3.0200e+02, 1.3020e+03]], |
| |
| [[1.2000e+01, 1.0120e+03], |
| [1.1200e+02, 1.1120e+03], |
| [2.1200e+02, 1.2120e+03], |
| [3.1200e+02, 1.3120e+03]], |
| |
| [[2.2000e+01, 1.0220e+03], |
| [1.2200e+02, 1.1220e+03], |
| [2.2200e+02, 1.2220e+03], |
| [3.2200e+02, 1.3220e+03]]], |
| |
| |
| [[[3.0000e+00, 1.0030e+03], |
| [1.0300e+02, 1.1030e+03], |
| [2.0300e+02, 1.2030e+03], |
| [3.0300e+02, 1.3030e+03]], |
| |
| [[1.3000e+01, 1.0130e+03], |
| [1.1300e+02, 1.1130e+03], |
| [2.1300e+02, 1.2130e+03], |
| [3.1300e+02, 1.3130e+03]], |
| |
| [[2.3000e+01, 1.0230e+03], |
| [1.2300e+02, 1.1230e+03], |
| [2.2300e+02, 1.2230e+03], |
| [3.2300e+02, 1.3230e+03]]]], |
| |
| |
| |
| [[[[3.0000e+00, 1.0030e+03], |
| [1.0300e+02, 1.1030e+03], |
| [2.0300e+02, 1.2030e+03], |
| [3.0300e+02, 1.3030e+03]], |
| |
| [[1.3000e+01, 1.0130e+03], |
| [1.1300e+02, 1.1130e+03], |
| [2.1300e+02, 1.2130e+03], |
| [3.1300e+02, 1.3130e+03]], |
| |
| [[2.3000e+01, 1.0230e+03], |
| [1.2300e+02, 1.1230e+03], |
| [2.2300e+02, 1.2230e+03], |
| [3.2300e+02, 1.3230e+03]]], |
| |
| |
| [[[4.0000e+00, 1.0040e+03], |
| [1.0400e+02, 1.1040e+03], |
| [2.0400e+02, 1.2040e+03], |
| [3.0400e+02, 1.3040e+03]], |
| |
| [[1.4000e+01, 1.0140e+03], |
| [1.1400e+02, 1.1140e+03], |
| [2.1400e+02, 1.2140e+03], |
| [3.1400e+02, 1.3140e+03]], |
| |
| [[2.4000e+01, 1.0240e+03], |
| [1.2400e+02, 1.1240e+03], |
| [2.2400e+02, 1.2240e+03], |
| [3.2400e+02, 1.3240e+03]]]], |
| |
| |
| |
| [[[[4.0000e+00, 1.0040e+03], |
| [1.0400e+02, 1.1040e+03], |
| [2.0400e+02, 1.2040e+03], |
| [3.0400e+02, 1.3040e+03]], |
| |
| [[1.4000e+01, 1.0140e+03], |
| [1.1400e+02, 1.1140e+03], |
| [2.1400e+02, 1.2140e+03], |
| [3.1400e+02, 1.3140e+03]], |
| |
| [[2.4000e+01, 1.0240e+03], |
| [1.2400e+02, 1.1240e+03], |
| [2.2400e+02, 1.2240e+03], |
| [3.2400e+02, 1.3240e+03]]], |
| |
| |
| [[[5.0000e+00, 1.0050e+03], |
| [1.0500e+02, 1.1050e+03], |
| [2.0500e+02, 1.2050e+03], |
| [3.0500e+02, 1.3050e+03]], |
| |
| [[1.5000e+01, 1.0150e+03], |
| [1.1500e+02, 1.1150e+03], |
| [2.1500e+02, 1.2150e+03], |
| [3.1500e+02, 1.3150e+03]], |
| |
| [[2.5000e+01, 1.0250e+03], |
| [1.2500e+02, 1.1250e+03], |
| [2.2500e+02, 1.2250e+03], |
| [3.2500e+02, 1.3250e+03]]]]], device='cuda:0') |
| |
| ########## torch.float32/torch.int64/size=(2, 3)+(6, 6)+(2, 1) ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([[[0, 2, 4], |
| [0, 3, 4], |
| [0, 1, 4]], |
| |
| [[0, 1, 4], |
| [0, 2, 4], |
| [0, 3, 4]]]), |
| col_indices=tensor([[[0, 1, 0, 1], |
| [0, 1, 2, 0], |
| [0, 0, 1, 2]], |
| |
| [[1, 0, 1, 2], |
| [0, 2, 0, 1], |
| [0, 1, 2, 1]]]), |
| values=tensor([[[[[[[ 1.], |
| [101.]], |
| |
| [[ 11.], |
| [111.]]], |
| |
| |
| [[[ 2.], |
| [102.]], |
| |
| [[ 12.], |
| [112.]]], |
| |
| |
| [[[ 3.], |
| [103.]], |
| |
| [[ 13.], |
| [113.]]]], |
| |
| |
| |
| [[[[ 2.], |
| [102.]], |
| |
| [[ 12.], |
| [112.]]], |
| |
| |
| [[[ 3.], |
| [103.]], |
| |
| [[ 13.], |
| [113.]]], |
| |
| |
| [[[ 4.], |
| [104.]], |
| |
| [[ 14.], |
| [114.]]]], |
| |
| |
| |
| [[[[ 3.], |
| [103.]], |
| |
| [[ 13.], |
| [113.]]], |
| |
| |
| [[[ 4.], |
| [104.]], |
| |
| [[ 14.], |
| [114.]]], |
| |
| |
| [[[ 5.], |
| [105.]], |
| |
| [[ 15.], |
| [115.]]]], |
| |
| |
| |
| [[[[ 4.], |
| [104.]], |
| |
| [[ 14.], |
| [114.]]], |
| |
| |
| [[[ 5.], |
| [105.]], |
| |
| [[ 15.], |
| [115.]]], |
| |
| |
| [[[ 6.], |
| [106.]], |
| |
| [[ 16.], |
| [116.]]]]], |
| |
| |
| |
| |
| [[[[[ 5.], |
| [105.]], |
| |
| [[ 15.], |
| [115.]]], |
| |
| |
| [[[ 6.], |
| [106.]], |
| |
| [[ 16.], |
| [116.]]], |
| |
| |
| [[[ 7.], |
| [107.]], |
| |
| [[ 17.], |
| [117.]]]], |
| |
| |
| |
| [[[[ 6.], |
| [106.]], |
| |
| [[ 16.], |
| [116.]]], |
| |
| |
| [[[ 7.], |
| [107.]], |
| |
| [[ 17.], |
| [117.]]], |
| |
| |
| [[[ 8.], |
| [108.]], |
| |
| [[ 18.], |
| [118.]]]], |
| |
| |
| |
| [[[[ 7.], |
| [107.]], |
| |
| [[ 17.], |
| [117.]]], |
| |
| |
| [[[ 8.], |
| [108.]], |
| |
| [[ 18.], |
| [118.]]], |
| |
| |
| [[[ 9.], |
| [109.]], |
| |
| [[ 19.], |
| [119.]]]], |
| |
| |
| |
| [[[[ 8.], |
| [108.]], |
| |
| [[ 18.], |
| [118.]]], |
| |
| |
| [[[ 9.], |
| [109.]], |
| |
| [[ 19.], |
| [119.]]], |
| |
| |
| [[[ 10.], |
| [110.]], |
| |
| [[ 20.], |
| [120.]]]]], |
| |
| |
| |
| |
| [[[[[ 9.], |
| [109.]], |
| |
| [[ 19.], |
| [119.]]], |
| |
| |
| [[[ 10.], |
| [110.]], |
| |
| [[ 20.], |
| [120.]]], |
| |
| |
| [[[ 11.], |
| [111.]], |
| |
| [[ 21.], |
| [121.]]]], |
| |
| |
| |
| [[[[ 10.], |
| [110.]], |
| |
| [[ 20.], |
| [120.]]], |
| |
| |
| [[[ 11.], |
| [111.]], |
| |
| [[ 21.], |
| [121.]]], |
| |
| |
| [[[ 12.], |
| [112.]], |
| |
| [[ 22.], |
| [122.]]]], |
| |
| |
| |
| [[[[ 11.], |
| [111.]], |
| |
| [[ 21.], |
| [121.]]], |
| |
| |
| [[[ 12.], |
| [112.]], |
| |
| [[ 22.], |
| [122.]]], |
| |
| |
| [[[ 13.], |
| [113.]], |
| |
| [[ 23.], |
| [123.]]]], |
| |
| |
| |
| [[[[ 12.], |
| [112.]], |
| |
| [[ 22.], |
| [122.]]], |
| |
| |
| [[[ 13.], |
| [113.]], |
| |
| [[ 23.], |
| [123.]]], |
| |
| |
| [[[ 14.], |
| [114.]], |
| |
| [[ 24.], |
| [124.]]]]]], |
| |
| |
| |
| |
| |
| [[[[[[ 13.], |
| [113.]], |
| |
| [[ 23.], |
| [123.]]], |
| |
| |
| [[[ 14.], |
| [114.]], |
| |
| [[ 24.], |
| [124.]]], |
| |
| |
| [[[ 15.], |
| [115.]], |
| |
| [[ 25.], |
| [125.]]]], |
| |
| |
| |
| [[[[ 14.], |
| [114.]], |
| |
| [[ 24.], |
| [124.]]], |
| |
| |
| [[[ 15.], |
| [115.]], |
| |
| [[ 25.], |
| [125.]]], |
| |
| |
| [[[ 16.], |
| [116.]], |
| |
| [[ 26.], |
| [126.]]]], |
| |
| |
| |
| [[[[ 15.], |
| [115.]], |
| |
| [[ 25.], |
| [125.]]], |
| |
| |
| [[[ 16.], |
| [116.]], |
| |
| [[ 26.], |
| [126.]]], |
| |
| |
| [[[ 17.], |
| [117.]], |
| |
| [[ 27.], |
| [127.]]]], |
| |
| |
| |
| [[[[ 16.], |
| [116.]], |
| |
| [[ 26.], |
| [126.]]], |
| |
| |
| [[[ 17.], |
| [117.]], |
| |
| [[ 27.], |
| [127.]]], |
| |
| |
| [[[ 18.], |
| [118.]], |
| |
| [[ 28.], |
| [128.]]]]], |
| |
| |
| |
| |
| [[[[[ 17.], |
| [117.]], |
| |
| [[ 27.], |
| [127.]]], |
| |
| |
| [[[ 18.], |
| [118.]], |
| |
| [[ 28.], |
| [128.]]], |
| |
| |
| [[[ 19.], |
| [119.]], |
| |
| [[ 29.], |
| [129.]]]], |
| |
| |
| |
| [[[[ 18.], |
| [118.]], |
| |
| [[ 28.], |
| [128.]]], |
| |
| |
| [[[ 19.], |
| [119.]], |
| |
| [[ 29.], |
| [129.]]], |
| |
| |
| [[[ 20.], |
| [120.]], |
| |
| [[ 30.], |
| [130.]]]], |
| |
| |
| |
| [[[[ 19.], |
| [119.]], |
| |
| [[ 29.], |
| [129.]]], |
| |
| |
| [[[ 20.], |
| [120.]], |
| |
| [[ 30.], |
| [130.]]], |
| |
| |
| [[[ 21.], |
| [121.]], |
| |
| [[ 31.], |
| [131.]]]], |
| |
| |
| |
| [[[[ 20.], |
| [120.]], |
| |
| [[ 30.], |
| [130.]]], |
| |
| |
| [[[ 21.], |
| [121.]], |
| |
| [[ 31.], |
| [131.]]], |
| |
| |
| [[[ 22.], |
| [122.]], |
| |
| [[ 32.], |
| [132.]]]]], |
| |
| |
| |
| |
| [[[[[ 21.], |
| [121.]], |
| |
| [[ 31.], |
| [131.]]], |
| |
| |
| [[[ 22.], |
| [122.]], |
| |
| [[ 32.], |
| [132.]]], |
| |
| |
| [[[ 23.], |
| [123.]], |
| |
| [[ 33.], |
| [133.]]]], |
| |
| |
| |
| [[[[ 22.], |
| [122.]], |
| |
| [[ 32.], |
| [132.]]], |
| |
| |
| [[[ 23.], |
| [123.]], |
| |
| [[ 33.], |
| [133.]]], |
| |
| |
| [[[ 24.], |
| [124.]], |
| |
| [[ 34.], |
| [134.]]]], |
| |
| |
| |
| [[[[ 23.], |
| [123.]], |
| |
| [[ 33.], |
| [133.]]], |
| |
| |
| [[[ 24.], |
| [124.]], |
| |
| [[ 34.], |
| [134.]]], |
| |
| |
| [[[ 25.], |
| [125.]], |
| |
| [[ 35.], |
| [135.]]]], |
| |
| |
| |
| [[[[ 24.], |
| [124.]], |
| |
| [[ 34.], |
| [134.]]], |
| |
| |
| [[[ 25.], |
| [125.]], |
| |
| [[ 35.], |
| [135.]]], |
| |
| |
| [[[ 26.], |
| [126.]], |
| |
| [[ 36.], |
| [136.]]]]]]]), device='cuda:0', |
| size=(2, 3, 6, 6, 2, 1), nnz=4, layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([[[0, 2, 4], |
| [0, 3, 4], |
| [0, 1, 4]], |
| |
| [[0, 1, 4], |
| [0, 2, 4], |
| [0, 3, 4]]], device='cuda:0') |
| # _col_indices |
| tensor([[[0, 1, 0, 1], |
| [0, 1, 2, 0], |
| [0, 0, 1, 2]], |
| |
| [[1, 0, 1, 2], |
| [0, 2, 0, 1], |
| [0, 1, 2, 1]]], device='cuda:0') |
| # _values |
| tensor([[[[[[[ 1.], |
| [101.]], |
| |
| [[ 11.], |
| [111.]]], |
| |
| |
| [[[ 2.], |
| [102.]], |
| |
| [[ 12.], |
| [112.]]], |
| |
| |
| [[[ 3.], |
| [103.]], |
| |
| [[ 13.], |
| [113.]]]], |
| |
| |
| |
| [[[[ 2.], |
| [102.]], |
| |
| [[ 12.], |
| [112.]]], |
| |
| |
| [[[ 3.], |
| [103.]], |
| |
| [[ 13.], |
| [113.]]], |
| |
| |
| [[[ 4.], |
| [104.]], |
| |
| [[ 14.], |
| [114.]]]], |
| |
| |
| |
| [[[[ 3.], |
| [103.]], |
| |
| [[ 13.], |
| [113.]]], |
| |
| |
| [[[ 4.], |
| [104.]], |
| |
| [[ 14.], |
| [114.]]], |
| |
| |
| [[[ 5.], |
| [105.]], |
| |
| [[ 15.], |
| [115.]]]], |
| |
| |
| |
| [[[[ 4.], |
| [104.]], |
| |
| [[ 14.], |
| [114.]]], |
| |
| |
| [[[ 5.], |
| [105.]], |
| |
| [[ 15.], |
| [115.]]], |
| |
| |
| [[[ 6.], |
| [106.]], |
| |
| [[ 16.], |
| [116.]]]]], |
| |
| |
| |
| |
| [[[[[ 5.], |
| [105.]], |
| |
| [[ 15.], |
| [115.]]], |
| |
| |
| [[[ 6.], |
| [106.]], |
| |
| [[ 16.], |
| [116.]]], |
| |
| |
| [[[ 7.], |
| [107.]], |
| |
| [[ 17.], |
| [117.]]]], |
| |
| |
| |
| [[[[ 6.], |
| [106.]], |
| |
| [[ 16.], |
| [116.]]], |
| |
| |
| [[[ 7.], |
| [107.]], |
| |
| [[ 17.], |
| [117.]]], |
| |
| |
| [[[ 8.], |
| [108.]], |
| |
| [[ 18.], |
| [118.]]]], |
| |
| |
| |
| [[[[ 7.], |
| [107.]], |
| |
| [[ 17.], |
| [117.]]], |
| |
| |
| [[[ 8.], |
| [108.]], |
| |
| [[ 18.], |
| [118.]]], |
| |
| |
| [[[ 9.], |
| [109.]], |
| |
| [[ 19.], |
| [119.]]]], |
| |
| |
| |
| [[[[ 8.], |
| [108.]], |
| |
| [[ 18.], |
| [118.]]], |
| |
| |
| [[[ 9.], |
| [109.]], |
| |
| [[ 19.], |
| [119.]]], |
| |
| |
| [[[ 10.], |
| [110.]], |
| |
| [[ 20.], |
| [120.]]]]], |
| |
| |
| |
| |
| [[[[[ 9.], |
| [109.]], |
| |
| [[ 19.], |
| [119.]]], |
| |
| |
| [[[ 10.], |
| [110.]], |
| |
| [[ 20.], |
| [120.]]], |
| |
| |
| [[[ 11.], |
| [111.]], |
| |
| [[ 21.], |
| [121.]]]], |
| |
| |
| |
| [[[[ 10.], |
| [110.]], |
| |
| [[ 20.], |
| [120.]]], |
| |
| |
| [[[ 11.], |
| [111.]], |
| |
| [[ 21.], |
| [121.]]], |
| |
| |
| [[[ 12.], |
| [112.]], |
| |
| [[ 22.], |
| [122.]]]], |
| |
| |
| |
| [[[[ 11.], |
| [111.]], |
| |
| [[ 21.], |
| [121.]]], |
| |
| |
| [[[ 12.], |
| [112.]], |
| |
| [[ 22.], |
| [122.]]], |
| |
| |
| [[[ 13.], |
| [113.]], |
| |
| [[ 23.], |
| [123.]]]], |
| |
| |
| |
| [[[[ 12.], |
| [112.]], |
| |
| [[ 22.], |
| [122.]]], |
| |
| |
| [[[ 13.], |
| [113.]], |
| |
| [[ 23.], |
| [123.]]], |
| |
| |
| [[[ 14.], |
| [114.]], |
| |
| [[ 24.], |
| [124.]]]]]], |
| |
| |
| |
| |
| |
| [[[[[[ 13.], |
| [113.]], |
| |
| [[ 23.], |
| [123.]]], |
| |
| |
| [[[ 14.], |
| [114.]], |
| |
| [[ 24.], |
| [124.]]], |
| |
| |
| [[[ 15.], |
| [115.]], |
| |
| [[ 25.], |
| [125.]]]], |
| |
| |
| |
| [[[[ 14.], |
| [114.]], |
| |
| [[ 24.], |
| [124.]]], |
| |
| |
| [[[ 15.], |
| [115.]], |
| |
| [[ 25.], |
| [125.]]], |
| |
| |
| [[[ 16.], |
| [116.]], |
| |
| [[ 26.], |
| [126.]]]], |
| |
| |
| |
| [[[[ 15.], |
| [115.]], |
| |
| [[ 25.], |
| [125.]]], |
| |
| |
| [[[ 16.], |
| [116.]], |
| |
| [[ 26.], |
| [126.]]], |
| |
| |
| [[[ 17.], |
| [117.]], |
| |
| [[ 27.], |
| [127.]]]], |
| |
| |
| |
| [[[[ 16.], |
| [116.]], |
| |
| [[ 26.], |
| [126.]]], |
| |
| |
| [[[ 17.], |
| [117.]], |
| |
| [[ 27.], |
| [127.]]], |
| |
| |
| [[[ 18.], |
| [118.]], |
| |
| [[ 28.], |
| [128.]]]]], |
| |
| |
| |
| |
| [[[[[ 17.], |
| [117.]], |
| |
| [[ 27.], |
| [127.]]], |
| |
| |
| [[[ 18.], |
| [118.]], |
| |
| [[ 28.], |
| [128.]]], |
| |
| |
| [[[ 19.], |
| [119.]], |
| |
| [[ 29.], |
| [129.]]]], |
| |
| |
| |
| [[[[ 18.], |
| [118.]], |
| |
| [[ 28.], |
| [128.]]], |
| |
| |
| [[[ 19.], |
| [119.]], |
| |
| [[ 29.], |
| [129.]]], |
| |
| |
| [[[ 20.], |
| [120.]], |
| |
| [[ 30.], |
| [130.]]]], |
| |
| |
| |
| [[[[ 19.], |
| [119.]], |
| |
| [[ 29.], |
| [129.]]], |
| |
| |
| [[[ 20.], |
| [120.]], |
| |
| [[ 30.], |
| [130.]]], |
| |
| |
| [[[ 21.], |
| [121.]], |
| |
| [[ 31.], |
| [131.]]]], |
| |
| |
| |
| [[[[ 20.], |
| [120.]], |
| |
| [[ 30.], |
| [130.]]], |
| |
| |
| [[[ 21.], |
| [121.]], |
| |
| [[ 31.], |
| [131.]]], |
| |
| |
| [[[ 22.], |
| [122.]], |
| |
| [[ 32.], |
| [132.]]]]], |
| |
| |
| |
| |
| [[[[[ 21.], |
| [121.]], |
| |
| [[ 31.], |
| [131.]]], |
| |
| |
| [[[ 22.], |
| [122.]], |
| |
| [[ 32.], |
| [132.]]], |
| |
| |
| [[[ 23.], |
| [123.]], |
| |
| [[ 33.], |
| [133.]]]], |
| |
| |
| |
| [[[[ 22.], |
| [122.]], |
| |
| [[ 32.], |
| [132.]]], |
| |
| |
| [[[ 23.], |
| [123.]], |
| |
| [[ 33.], |
| [133.]]], |
| |
| |
| [[[ 24.], |
| [124.]], |
| |
| [[ 34.], |
| [134.]]]], |
| |
| |
| |
| [[[[ 23.], |
| [123.]], |
| |
| [[ 33.], |
| [133.]]], |
| |
| |
| [[[ 24.], |
| [124.]], |
| |
| [[ 34.], |
| [134.]]], |
| |
| |
| [[[ 25.], |
| [125.]], |
| |
| [[ 35.], |
| [135.]]]], |
| |
| |
| |
| [[[[ 24.], |
| [124.]], |
| |
| [[ 34.], |
| [134.]]], |
| |
| |
| [[[ 25.], |
| [125.]], |
| |
| [[ 35.], |
| [135.]]], |
| |
| |
| [[[ 26.], |
| [126.]], |
| |
| [[ 36.], |
| [136.]]]]]]], device='cuda:0') |
| |
| |
| ########## torch.float64/torch.int64/size=()+(6, 6)+(2,) ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([0, 2, 4]), |
| col_indices=tensor([0, 1, 0, 2]), |
| values=tensor([[[[ 1., 101.], |
| [ 11., 111.]], |
| |
| [[ 2., 102.], |
| [ 12., 112.]], |
| |
| [[ 3., 103.], |
| [ 13., 113.]]], |
| |
| |
| [[[ 2., 102.], |
| [ 12., 112.]], |
| |
| [[ 3., 103.], |
| [ 13., 113.]], |
| |
| [[ 4., 104.], |
| [ 14., 114.]]], |
| |
| |
| [[[ 3., 103.], |
| [ 13., 113.]], |
| |
| [[ 4., 104.], |
| [ 14., 114.]], |
| |
| [[ 5., 105.], |
| [ 15., 115.]]], |
| |
| |
| [[[ 4., 104.], |
| [ 14., 114.]], |
| |
| [[ 5., 105.], |
| [ 15., 115.]], |
| |
| [[ 6., 106.], |
| [ 16., 116.]]]]), device='cuda:0', size=(6, 6, 2), |
| nnz=4, dtype=torch.float64, layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([0, 2, 4], device='cuda:0') |
| # _col_indices |
| tensor([0, 1, 0, 2], device='cuda:0') |
| # _values |
| tensor([[[[ 1., 101.], |
| [ 11., 111.]], |
| |
| [[ 2., 102.], |
| [ 12., 112.]], |
| |
| [[ 3., 103.], |
| [ 13., 113.]]], |
| |
| |
| [[[ 2., 102.], |
| [ 12., 112.]], |
| |
| [[ 3., 103.], |
| [ 13., 113.]], |
| |
| [[ 4., 104.], |
| [ 14., 114.]]], |
| |
| |
| [[[ 3., 103.], |
| [ 13., 113.]], |
| |
| [[ 4., 104.], |
| [ 14., 114.]], |
| |
| [[ 5., 105.], |
| [ 15., 115.]]], |
| |
| |
| [[[ 4., 104.], |
| [ 14., 114.]], |
| |
| [[ 5., 105.], |
| [ 15., 115.]], |
| |
| [[ 6., 106.], |
| [ 16., 116.]]]], device='cuda:0', dtype=torch.float64) |
| |
| ########## torch.float64/torch.int64/size=()+(4, 9)+(4, 2) ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([0, 2, 4]), |
| col_indices=tensor([0, 1, 0, 2]), |
| values=tensor([[[[[1.0000e+00, 1.0010e+03], |
| [1.0100e+02, 1.1010e+03], |
| [2.0100e+02, 1.2010e+03], |
| [3.0100e+02, 1.3010e+03]], |
| |
| [[1.1000e+01, 1.0110e+03], |
| [1.1100e+02, 1.1110e+03], |
| [2.1100e+02, 1.2110e+03], |
| [3.1100e+02, 1.3110e+03]], |
| |
| [[2.1000e+01, 1.0210e+03], |
| [1.2100e+02, 1.1210e+03], |
| [2.2100e+02, 1.2210e+03], |
| [3.2100e+02, 1.3210e+03]]], |
| |
| |
| [[[2.0000e+00, 1.0020e+03], |
| [1.0200e+02, 1.1020e+03], |
| [2.0200e+02, 1.2020e+03], |
| [3.0200e+02, 1.3020e+03]], |
| |
| [[1.2000e+01, 1.0120e+03], |
| [1.1200e+02, 1.1120e+03], |
| [2.1200e+02, 1.2120e+03], |
| [3.1200e+02, 1.3120e+03]], |
| |
| [[2.2000e+01, 1.0220e+03], |
| [1.2200e+02, 1.1220e+03], |
| [2.2200e+02, 1.2220e+03], |
| [3.2200e+02, 1.3220e+03]]]], |
| |
| |
| |
| [[[[2.0000e+00, 1.0020e+03], |
| [1.0200e+02, 1.1020e+03], |
| [2.0200e+02, 1.2020e+03], |
| [3.0200e+02, 1.3020e+03]], |
| |
| [[1.2000e+01, 1.0120e+03], |
| [1.1200e+02, 1.1120e+03], |
| [2.1200e+02, 1.2120e+03], |
| [3.1200e+02, 1.3120e+03]], |
| |
| [[2.2000e+01, 1.0220e+03], |
| [1.2200e+02, 1.1220e+03], |
| [2.2200e+02, 1.2220e+03], |
| [3.2200e+02, 1.3220e+03]]], |
| |
| |
| [[[3.0000e+00, 1.0030e+03], |
| [1.0300e+02, 1.1030e+03], |
| [2.0300e+02, 1.2030e+03], |
| [3.0300e+02, 1.3030e+03]], |
| |
| [[1.3000e+01, 1.0130e+03], |
| [1.1300e+02, 1.1130e+03], |
| [2.1300e+02, 1.2130e+03], |
| [3.1300e+02, 1.3130e+03]], |
| |
| [[2.3000e+01, 1.0230e+03], |
| [1.2300e+02, 1.1230e+03], |
| [2.2300e+02, 1.2230e+03], |
| [3.2300e+02, 1.3230e+03]]]], |
| |
| |
| |
| [[[[3.0000e+00, 1.0030e+03], |
| [1.0300e+02, 1.1030e+03], |
| [2.0300e+02, 1.2030e+03], |
| [3.0300e+02, 1.3030e+03]], |
| |
| [[1.3000e+01, 1.0130e+03], |
| [1.1300e+02, 1.1130e+03], |
| [2.1300e+02, 1.2130e+03], |
| [3.1300e+02, 1.3130e+03]], |
| |
| [[2.3000e+01, 1.0230e+03], |
| [1.2300e+02, 1.1230e+03], |
| [2.2300e+02, 1.2230e+03], |
| [3.2300e+02, 1.3230e+03]]], |
| |
| |
| [[[4.0000e+00, 1.0040e+03], |
| [1.0400e+02, 1.1040e+03], |
| [2.0400e+02, 1.2040e+03], |
| [3.0400e+02, 1.3040e+03]], |
| |
| [[1.4000e+01, 1.0140e+03], |
| [1.1400e+02, 1.1140e+03], |
| [2.1400e+02, 1.2140e+03], |
| [3.1400e+02, 1.3140e+03]], |
| |
| [[2.4000e+01, 1.0240e+03], |
| [1.2400e+02, 1.1240e+03], |
| [2.2400e+02, 1.2240e+03], |
| [3.2400e+02, 1.3240e+03]]]], |
| |
| |
| |
| [[[[4.0000e+00, 1.0040e+03], |
| [1.0400e+02, 1.1040e+03], |
| [2.0400e+02, 1.2040e+03], |
| [3.0400e+02, 1.3040e+03]], |
| |
| [[1.4000e+01, 1.0140e+03], |
| [1.1400e+02, 1.1140e+03], |
| [2.1400e+02, 1.2140e+03], |
| [3.1400e+02, 1.3140e+03]], |
| |
| [[2.4000e+01, 1.0240e+03], |
| [1.2400e+02, 1.1240e+03], |
| [2.2400e+02, 1.2240e+03], |
| [3.2400e+02, 1.3240e+03]]], |
| |
| |
| [[[5.0000e+00, 1.0050e+03], |
| [1.0500e+02, 1.1050e+03], |
| [2.0500e+02, 1.2050e+03], |
| [3.0500e+02, 1.3050e+03]], |
| |
| [[1.5000e+01, 1.0150e+03], |
| [1.1500e+02, 1.1150e+03], |
| [2.1500e+02, 1.2150e+03], |
| [3.1500e+02, 1.3150e+03]], |
| |
| [[2.5000e+01, 1.0250e+03], |
| [1.2500e+02, 1.1250e+03], |
| [2.2500e+02, 1.2250e+03], |
| [3.2500e+02, 1.3250e+03]]]]]), device='cuda:0', |
| size=(4, 9, 4, 2), nnz=4, dtype=torch.float64, layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([0, 2, 4], device='cuda:0') |
| # _col_indices |
| tensor([0, 1, 0, 2], device='cuda:0') |
| # _values |
| tensor([[[[[1.0000e+00, 1.0010e+03], |
| [1.0100e+02, 1.1010e+03], |
| [2.0100e+02, 1.2010e+03], |
| [3.0100e+02, 1.3010e+03]], |
| |
| [[1.1000e+01, 1.0110e+03], |
| [1.1100e+02, 1.1110e+03], |
| [2.1100e+02, 1.2110e+03], |
| [3.1100e+02, 1.3110e+03]], |
| |
| [[2.1000e+01, 1.0210e+03], |
| [1.2100e+02, 1.1210e+03], |
| [2.2100e+02, 1.2210e+03], |
| [3.2100e+02, 1.3210e+03]]], |
| |
| |
| [[[2.0000e+00, 1.0020e+03], |
| [1.0200e+02, 1.1020e+03], |
| [2.0200e+02, 1.2020e+03], |
| [3.0200e+02, 1.3020e+03]], |
| |
| [[1.2000e+01, 1.0120e+03], |
| [1.1200e+02, 1.1120e+03], |
| [2.1200e+02, 1.2120e+03], |
| [3.1200e+02, 1.3120e+03]], |
| |
| [[2.2000e+01, 1.0220e+03], |
| [1.2200e+02, 1.1220e+03], |
| [2.2200e+02, 1.2220e+03], |
| [3.2200e+02, 1.3220e+03]]]], |
| |
| |
| |
| [[[[2.0000e+00, 1.0020e+03], |
| [1.0200e+02, 1.1020e+03], |
| [2.0200e+02, 1.2020e+03], |
| [3.0200e+02, 1.3020e+03]], |
| |
| [[1.2000e+01, 1.0120e+03], |
| [1.1200e+02, 1.1120e+03], |
| [2.1200e+02, 1.2120e+03], |
| [3.1200e+02, 1.3120e+03]], |
| |
| [[2.2000e+01, 1.0220e+03], |
| [1.2200e+02, 1.1220e+03], |
| [2.2200e+02, 1.2220e+03], |
| [3.2200e+02, 1.3220e+03]]], |
| |
| |
| [[[3.0000e+00, 1.0030e+03], |
| [1.0300e+02, 1.1030e+03], |
| [2.0300e+02, 1.2030e+03], |
| [3.0300e+02, 1.3030e+03]], |
| |
| [[1.3000e+01, 1.0130e+03], |
| [1.1300e+02, 1.1130e+03], |
| [2.1300e+02, 1.2130e+03], |
| [3.1300e+02, 1.3130e+03]], |
| |
| [[2.3000e+01, 1.0230e+03], |
| [1.2300e+02, 1.1230e+03], |
| [2.2300e+02, 1.2230e+03], |
| [3.2300e+02, 1.3230e+03]]]], |
| |
| |
| |
| [[[[3.0000e+00, 1.0030e+03], |
| [1.0300e+02, 1.1030e+03], |
| [2.0300e+02, 1.2030e+03], |
| [3.0300e+02, 1.3030e+03]], |
| |
| [[1.3000e+01, 1.0130e+03], |
| [1.1300e+02, 1.1130e+03], |
| [2.1300e+02, 1.2130e+03], |
| [3.1300e+02, 1.3130e+03]], |
| |
| [[2.3000e+01, 1.0230e+03], |
| [1.2300e+02, 1.1230e+03], |
| [2.2300e+02, 1.2230e+03], |
| [3.2300e+02, 1.3230e+03]]], |
| |
| |
| [[[4.0000e+00, 1.0040e+03], |
| [1.0400e+02, 1.1040e+03], |
| [2.0400e+02, 1.2040e+03], |
| [3.0400e+02, 1.3040e+03]], |
| |
| [[1.4000e+01, 1.0140e+03], |
| [1.1400e+02, 1.1140e+03], |
| [2.1400e+02, 1.2140e+03], |
| [3.1400e+02, 1.3140e+03]], |
| |
| [[2.4000e+01, 1.0240e+03], |
| [1.2400e+02, 1.1240e+03], |
| [2.2400e+02, 1.2240e+03], |
| [3.2400e+02, 1.3240e+03]]]], |
| |
| |
| |
| [[[[4.0000e+00, 1.0040e+03], |
| [1.0400e+02, 1.1040e+03], |
| [2.0400e+02, 1.2040e+03], |
| [3.0400e+02, 1.3040e+03]], |
| |
| [[1.4000e+01, 1.0140e+03], |
| [1.1400e+02, 1.1140e+03], |
| [2.1400e+02, 1.2140e+03], |
| [3.1400e+02, 1.3140e+03]], |
| |
| [[2.4000e+01, 1.0240e+03], |
| [1.2400e+02, 1.1240e+03], |
| [2.2400e+02, 1.2240e+03], |
| [3.2400e+02, 1.3240e+03]]], |
| |
| |
| [[[5.0000e+00, 1.0050e+03], |
| [1.0500e+02, 1.1050e+03], |
| [2.0500e+02, 1.2050e+03], |
| [3.0500e+02, 1.3050e+03]], |
| |
| [[1.5000e+01, 1.0150e+03], |
| [1.1500e+02, 1.1150e+03], |
| [2.1500e+02, 1.2150e+03], |
| [3.1500e+02, 1.3150e+03]], |
| |
| [[2.5000e+01, 1.0250e+03], |
| [1.2500e+02, 1.1250e+03], |
| [2.2500e+02, 1.2250e+03], |
| [3.2500e+02, 1.3250e+03]]]]], device='cuda:0', dtype=torch.float64) |
| |
| ########## torch.float64/torch.int64/size=(2, 3)+(6, 6)+(2, 1) ########## |
| # sparse tensor |
| tensor(crow_indices=tensor([[[0, 2, 4], |
| [0, 3, 4], |
| [0, 1, 4]], |
| |
| [[0, 1, 4], |
| [0, 2, 4], |
| [0, 3, 4]]]), |
| col_indices=tensor([[[0, 1, 0, 1], |
| [0, 1, 2, 0], |
| [0, 0, 1, 2]], |
| |
| [[1, 0, 1, 2], |
| [0, 2, 0, 1], |
| [0, 1, 2, 1]]]), |
| values=tensor([[[[[[[ 1.], |
| [101.]], |
| |
| [[ 11.], |
| [111.]]], |
| |
| |
| [[[ 2.], |
| [102.]], |
| |
| [[ 12.], |
| [112.]]], |
| |
| |
| [[[ 3.], |
| [103.]], |
| |
| [[ 13.], |
| [113.]]]], |
| |
| |
| |
| [[[[ 2.], |
| [102.]], |
| |
| [[ 12.], |
| [112.]]], |
| |
| |
| [[[ 3.], |
| [103.]], |
| |
| [[ 13.], |
| [113.]]], |
| |
| |
| [[[ 4.], |
| [104.]], |
| |
| [[ 14.], |
| [114.]]]], |
| |
| |
| |
| [[[[ 3.], |
| [103.]], |
| |
| [[ 13.], |
| [113.]]], |
| |
| |
| [[[ 4.], |
| [104.]], |
| |
| [[ 14.], |
| [114.]]], |
| |
| |
| [[[ 5.], |
| [105.]], |
| |
| [[ 15.], |
| [115.]]]], |
| |
| |
| |
| [[[[ 4.], |
| [104.]], |
| |
| [[ 14.], |
| [114.]]], |
| |
| |
| [[[ 5.], |
| [105.]], |
| |
| [[ 15.], |
| [115.]]], |
| |
| |
| [[[ 6.], |
| [106.]], |
| |
| [[ 16.], |
| [116.]]]]], |
| |
| |
| |
| |
| [[[[[ 5.], |
| [105.]], |
| |
| [[ 15.], |
| [115.]]], |
| |
| |
| [[[ 6.], |
| [106.]], |
| |
| [[ 16.], |
| [116.]]], |
| |
| |
| [[[ 7.], |
| [107.]], |
| |
| [[ 17.], |
| [117.]]]], |
| |
| |
| |
| [[[[ 6.], |
| [106.]], |
| |
| [[ 16.], |
| [116.]]], |
| |
| |
| [[[ 7.], |
| [107.]], |
| |
| [[ 17.], |
| [117.]]], |
| |
| |
| [[[ 8.], |
| [108.]], |
| |
| [[ 18.], |
| [118.]]]], |
| |
| |
| |
| [[[[ 7.], |
| [107.]], |
| |
| [[ 17.], |
| [117.]]], |
| |
| |
| [[[ 8.], |
| [108.]], |
| |
| [[ 18.], |
| [118.]]], |
| |
| |
| [[[ 9.], |
| [109.]], |
| |
| [[ 19.], |
| [119.]]]], |
| |
| |
| |
| [[[[ 8.], |
| [108.]], |
| |
| [[ 18.], |
| [118.]]], |
| |
| |
| [[[ 9.], |
| [109.]], |
| |
| [[ 19.], |
| [119.]]], |
| |
| |
| [[[ 10.], |
| [110.]], |
| |
| [[ 20.], |
| [120.]]]]], |
| |
| |
| |
| |
| [[[[[ 9.], |
| [109.]], |
| |
| [[ 19.], |
| [119.]]], |
| |
| |
| [[[ 10.], |
| [110.]], |
| |
| [[ 20.], |
| [120.]]], |
| |
| |
| [[[ 11.], |
| [111.]], |
| |
| [[ 21.], |
| [121.]]]], |
| |
| |
| |
| [[[[ 10.], |
| [110.]], |
| |
| [[ 20.], |
| [120.]]], |
| |
| |
| [[[ 11.], |
| [111.]], |
| |
| [[ 21.], |
| [121.]]], |
| |
| |
| [[[ 12.], |
| [112.]], |
| |
| [[ 22.], |
| [122.]]]], |
| |
| |
| |
| [[[[ 11.], |
| [111.]], |
| |
| [[ 21.], |
| [121.]]], |
| |
| |
| [[[ 12.], |
| [112.]], |
| |
| [[ 22.], |
| [122.]]], |
| |
| |
| [[[ 13.], |
| [113.]], |
| |
| [[ 23.], |
| [123.]]]], |
| |
| |
| |
| [[[[ 12.], |
| [112.]], |
| |
| [[ 22.], |
| [122.]]], |
| |
| |
| [[[ 13.], |
| [113.]], |
| |
| [[ 23.], |
| [123.]]], |
| |
| |
| [[[ 14.], |
| [114.]], |
| |
| [[ 24.], |
| [124.]]]]]], |
| |
| |
| |
| |
| |
| [[[[[[ 13.], |
| [113.]], |
| |
| [[ 23.], |
| [123.]]], |
| |
| |
| [[[ 14.], |
| [114.]], |
| |
| [[ 24.], |
| [124.]]], |
| |
| |
| [[[ 15.], |
| [115.]], |
| |
| [[ 25.], |
| [125.]]]], |
| |
| |
| |
| [[[[ 14.], |
| [114.]], |
| |
| [[ 24.], |
| [124.]]], |
| |
| |
| [[[ 15.], |
| [115.]], |
| |
| [[ 25.], |
| [125.]]], |
| |
| |
| [[[ 16.], |
| [116.]], |
| |
| [[ 26.], |
| [126.]]]], |
| |
| |
| |
| [[[[ 15.], |
| [115.]], |
| |
| [[ 25.], |
| [125.]]], |
| |
| |
| [[[ 16.], |
| [116.]], |
| |
| [[ 26.], |
| [126.]]], |
| |
| |
| [[[ 17.], |
| [117.]], |
| |
| [[ 27.], |
| [127.]]]], |
| |
| |
| |
| [[[[ 16.], |
| [116.]], |
| |
| [[ 26.], |
| [126.]]], |
| |
| |
| [[[ 17.], |
| [117.]], |
| |
| [[ 27.], |
| [127.]]], |
| |
| |
| [[[ 18.], |
| [118.]], |
| |
| [[ 28.], |
| [128.]]]]], |
| |
| |
| |
| |
| [[[[[ 17.], |
| [117.]], |
| |
| [[ 27.], |
| [127.]]], |
| |
| |
| [[[ 18.], |
| [118.]], |
| |
| [[ 28.], |
| [128.]]], |
| |
| |
| [[[ 19.], |
| [119.]], |
| |
| [[ 29.], |
| [129.]]]], |
| |
| |
| |
| [[[[ 18.], |
| [118.]], |
| |
| [[ 28.], |
| [128.]]], |
| |
| |
| [[[ 19.], |
| [119.]], |
| |
| [[ 29.], |
| [129.]]], |
| |
| |
| [[[ 20.], |
| [120.]], |
| |
| [[ 30.], |
| [130.]]]], |
| |
| |
| |
| [[[[ 19.], |
| [119.]], |
| |
| [[ 29.], |
| [129.]]], |
| |
| |
| [[[ 20.], |
| [120.]], |
| |
| [[ 30.], |
| [130.]]], |
| |
| |
| [[[ 21.], |
| [121.]], |
| |
| [[ 31.], |
| [131.]]]], |
| |
| |
| |
| [[[[ 20.], |
| [120.]], |
| |
| [[ 30.], |
| [130.]]], |
| |
| |
| [[[ 21.], |
| [121.]], |
| |
| [[ 31.], |
| [131.]]], |
| |
| |
| [[[ 22.], |
| [122.]], |
| |
| [[ 32.], |
| [132.]]]]], |
| |
| |
| |
| |
| [[[[[ 21.], |
| [121.]], |
| |
| [[ 31.], |
| [131.]]], |
| |
| |
| [[[ 22.], |
| [122.]], |
| |
| [[ 32.], |
| [132.]]], |
| |
| |
| [[[ 23.], |
| [123.]], |
| |
| [[ 33.], |
| [133.]]]], |
| |
| |
| |
| [[[[ 22.], |
| [122.]], |
| |
| [[ 32.], |
| [132.]]], |
| |
| |
| [[[ 23.], |
| [123.]], |
| |
| [[ 33.], |
| [133.]]], |
| |
| |
| [[[ 24.], |
| [124.]], |
| |
| [[ 34.], |
| [134.]]]], |
| |
| |
| |
| [[[[ 23.], |
| [123.]], |
| |
| [[ 33.], |
| [133.]]], |
| |
| |
| [[[ 24.], |
| [124.]], |
| |
| [[ 34.], |
| [134.]]], |
| |
| |
| [[[ 25.], |
| [125.]], |
| |
| [[ 35.], |
| [135.]]]], |
| |
| |
| |
| [[[[ 24.], |
| [124.]], |
| |
| [[ 34.], |
| [134.]]], |
| |
| |
| [[[ 25.], |
| [125.]], |
| |
| [[ 35.], |
| [135.]]], |
| |
| |
| [[[ 26.], |
| [126.]], |
| |
| [[ 36.], |
| [136.]]]]]]]), device='cuda:0', |
| size=(2, 3, 6, 6, 2, 1), nnz=4, dtype=torch.float64, |
| layout=torch.sparse_bsr) |
| # _crow_indices |
| tensor([[[0, 2, 4], |
| [0, 3, 4], |
| [0, 1, 4]], |
| |
| [[0, 1, 4], |
| [0, 2, 4], |
| [0, 3, 4]]], device='cuda:0') |
| # _col_indices |
| tensor([[[0, 1, 0, 1], |
| [0, 1, 2, 0], |
| [0, 0, 1, 2]], |
| |
| [[1, 0, 1, 2], |
| [0, 2, 0, 1], |
| [0, 1, 2, 1]]], device='cuda:0') |
| # _values |
| tensor([[[[[[[ 1.], |
| [101.]], |
| |
| [[ 11.], |
| [111.]]], |
| |
| |
| [[[ 2.], |
| [102.]], |
| |
| [[ 12.], |
| [112.]]], |
| |
| |
| [[[ 3.], |
| [103.]], |
| |
| [[ 13.], |
| [113.]]]], |
| |
| |
| |
| [[[[ 2.], |
| [102.]], |
| |
| [[ 12.], |
| [112.]]], |
| |
| |
| [[[ 3.], |
| [103.]], |
| |
| [[ 13.], |
| [113.]]], |
| |
| |
| [[[ 4.], |
| [104.]], |
| |
| [[ 14.], |
| [114.]]]], |
| |
| |
| |
| [[[[ 3.], |
| [103.]], |
| |
| [[ 13.], |
| [113.]]], |
| |
| |
| [[[ 4.], |
| [104.]], |
| |
| [[ 14.], |
| [114.]]], |
| |
| |
| [[[ 5.], |
| [105.]], |
| |
| [[ 15.], |
| [115.]]]], |
| |
| |
| |
| [[[[ 4.], |
| [104.]], |
| |
| [[ 14.], |
| [114.]]], |
| |
| |
| [[[ 5.], |
| [105.]], |
| |
| [[ 15.], |
| [115.]]], |
| |
| |
| [[[ 6.], |
| [106.]], |
| |
| [[ 16.], |
| [116.]]]]], |
| |
| |
| |
| |
| [[[[[ 5.], |
| [105.]], |
| |
| [[ 15.], |
| [115.]]], |
| |
| |
| [[[ 6.], |
| [106.]], |
| |
| [[ 16.], |
| [116.]]], |
| |
| |
| [[[ 7.], |
| [107.]], |
| |
| [[ 17.], |
| [117.]]]], |
| |
| |
| |
| [[[[ 6.], |
| [106.]], |
| |
| [[ 16.], |
| [116.]]], |
| |
| |
| [[[ 7.], |
| [107.]], |
| |
| [[ 17.], |
| [117.]]], |
| |
| |
| [[[ 8.], |
| [108.]], |
| |
| [[ 18.], |
| [118.]]]], |
| |
| |
| |
| [[[[ 7.], |
| [107.]], |
| |
| [[ 17.], |
| [117.]]], |
| |
| |
| [[[ 8.], |
| [108.]], |
| |
| [[ 18.], |
| [118.]]], |
| |
| |
| [[[ 9.], |
| [109.]], |
| |
| [[ 19.], |
| [119.]]]], |
| |
| |
| |
| [[[[ 8.], |
| [108.]], |
| |
| [[ 18.], |
| [118.]]], |
| |
| |
| [[[ 9.], |
| [109.]], |
| |
| [[ 19.], |
| [119.]]], |
| |
| |
| [[[ 10.], |
| [110.]], |
| |
| [[ 20.], |
| [120.]]]]], |
| |
| |
| |
| |
| [[[[[ 9.], |
| [109.]], |
| |
| [[ 19.], |
| [119.]]], |
| |
| |
| [[[ 10.], |
| [110.]], |
| |
| [[ 20.], |
| [120.]]], |
| |
| |
| [[[ 11.], |
| [111.]], |
| |
| [[ 21.], |
| [121.]]]], |
| |
| |
| |
| [[[[ 10.], |
| [110.]], |
| |
| [[ 20.], |
| [120.]]], |
| |
| |
| [[[ 11.], |
| [111.]], |
| |
| [[ 21.], |
| [121.]]], |
| |
| |
| [[[ 12.], |
| [112.]], |
| |
| [[ 22.], |
| [122.]]]], |
| |
| |
| |
| [[[[ 11.], |
| [111.]], |
| |
| [[ 21.], |
| [121.]]], |
| |
| |
| [[[ 12.], |
| [112.]], |
| |
| [[ 22.], |
| [122.]]], |
| |
| |
| [[[ 13.], |
| [113.]], |
| |
| [[ 23.], |
| [123.]]]], |
| |
| |
| |
| [[[[ 12.], |
| [112.]], |
| |
| [[ 22.], |
| [122.]]], |
| |
| |
| [[[ 13.], |
| [113.]], |
| |
| [[ 23.], |
| [123.]]], |
| |
| |
| [[[ 14.], |
| [114.]], |
| |
| [[ 24.], |
| [124.]]]]]], |
| |
| |
| |
| |
| |
| [[[[[[ 13.], |
| [113.]], |
| |
| [[ 23.], |
| [123.]]], |
| |
| |
| [[[ 14.], |
| [114.]], |
| |
| [[ 24.], |
| [124.]]], |
| |
| |
| [[[ 15.], |
| [115.]], |
| |
| [[ 25.], |
| [125.]]]], |
| |
| |
| |
| [[[[ 14.], |
| [114.]], |
| |
| [[ 24.], |
| [124.]]], |
| |
| |
| [[[ 15.], |
| [115.]], |
| |
| [[ 25.], |
| [125.]]], |
| |
| |
| [[[ 16.], |
| [116.]], |
| |
| [[ 26.], |
| [126.]]]], |
| |
| |
| |
| [[[[ 15.], |
| [115.]], |
| |
| [[ 25.], |
| [125.]]], |
| |
| |
| [[[ 16.], |
| [116.]], |
| |
| [[ 26.], |
| [126.]]], |
| |
| |
| [[[ 17.], |
| [117.]], |
| |
| [[ 27.], |
| [127.]]]], |
| |
| |
| |
| [[[[ 16.], |
| [116.]], |
| |
| [[ 26.], |
| [126.]]], |
| |
| |
| [[[ 17.], |
| [117.]], |
| |
| [[ 27.], |
| [127.]]], |
| |
| |
| [[[ 18.], |
| [118.]], |
| |
| [[ 28.], |
| [128.]]]]], |
| |
| |
| |
| |
| [[[[[ 17.], |
| [117.]], |
| |
| [[ 27.], |
| [127.]]], |
| |
| |
| [[[ 18.], |
| [118.]], |
| |
| [[ 28.], |
| [128.]]], |
| |
| |
| [[[ 19.], |
| [119.]], |
| |
| [[ 29.], |
| [129.]]]], |
| |
| |
| |
| [[[[ 18.], |
| [118.]], |
| |
| [[ 28.], |
| [128.]]], |
| |
| |
| [[[ 19.], |
| [119.]], |
| |
| [[ 29.], |
| [129.]]], |
| |
| |
| [[[ 20.], |
| [120.]], |
| |
| [[ 30.], |
| [130.]]]], |
| |
| |
| |
| [[[[ 19.], |
| [119.]], |
| |
| [[ 29.], |
| [129.]]], |
| |
| |
| [[[ 20.], |
| [120.]], |
| |
| [[ 30.], |
| [130.]]], |
| |
| |
| [[[ 21.], |
| [121.]], |
| |
| [[ 31.], |
| [131.]]]], |
| |
| |
| |
| [[[[ 20.], |
| [120.]], |
| |
| [[ 30.], |
| [130.]]], |
| |
| |
| [[[ 21.], |
| [121.]], |
| |
| [[ 31.], |
| [131.]]], |
| |
| |
| [[[ 22.], |
| [122.]], |
| |
| [[ 32.], |
| [132.]]]]], |
| |
| |
| |
| |
| [[[[[ 21.], |
| [121.]], |
| |
| [[ 31.], |
| [131.]]], |
| |
| |
| [[[ 22.], |
| [122.]], |
| |
| [[ 32.], |
| [132.]]], |
| |
| |
| [[[ 23.], |
| [123.]], |
| |
| [[ 33.], |
| [133.]]]], |
| |
| |
| |
| [[[[ 22.], |
| [122.]], |
| |
| [[ 32.], |
| [132.]]], |
| |
| |
| [[[ 23.], |
| [123.]], |
| |
| [[ 33.], |
| [133.]]], |
| |
| |
| [[[ 24.], |
| [124.]], |
| |
| [[ 34.], |
| [134.]]]], |
| |
| |
| |
| [[[[ 23.], |
| [123.]], |
| |
| [[ 33.], |
| [133.]]], |
| |
| |
| [[[ 24.], |
| [124.]], |
| |
| [[ 34.], |
| [134.]]], |
| |
| |
| [[[ 25.], |
| [125.]], |
| |
| [[ 35.], |
| [135.]]]], |
| |
| |
| |
| [[[[ 24.], |
| [124.]], |
| |
| [[ 34.], |
| [134.]]], |
| |
| |
| [[[ 25.], |
| [125.]], |
| |
| [[ 35.], |
| [135.]]], |
| |
| |
| [[[ 26.], |
| [126.]], |
| |
| [[ 36.], |
| [136.]]]]]]], device='cuda:0', dtype=torch.float64) |
| |
