| ########## torch.float32/torch.int32/size=()+(3, 2)+() ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([0, 2, 4]), | |
| row_indices=tensor([0, 1, 0, 2]), | |
| values=tensor([1., 2., 3., 4.]), device='cuda:0', size=(3, 2), nnz=4, | |
| layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0, 2, 4], device='cuda:0', dtype=torch.int32) | |
| # _row_indices | |
| tensor([0, 1, 0, 2], device='cuda:0', dtype=torch.int32) | |
| # _values | |
| tensor([1., 2., 3., 4.], device='cuda:0') | |
| ########## torch.float32/torch.int32/size=()+(0, 0)+() ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([0]), | |
| row_indices=tensor([], size=(0,)), | |
| values=tensor([], size=(0,)), device='cuda:0', size=(0, 0), nnz=0, | |
| layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0], device='cuda:0', dtype=torch.int32) | |
| # _row_indices | |
| tensor([], device='cuda:0', dtype=torch.int32) | |
| # _values | |
| tensor([], device='cuda:0') | |
| ########## torch.float32/torch.int32/size=(2,)+(3, 2)+() ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([[0, 2, 4], | |
| [0, 3, 4]]), | |
| row_indices=tensor([[0, 1, 0, 1], | |
| [0, 1, 2, 0]]), | |
| values=tensor([[1., 2., 3., 4.], | |
| [5., 6., 7., 8.]]), device='cuda:0', size=(2, 3, 2), | |
| nnz=4, layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([[0, 2, 4], | |
| [0, 3, 4]], device='cuda:0', dtype=torch.int32) | |
| # _row_indices | |
| tensor([[0, 1, 0, 1], | |
| [0, 1, 2, 0]], device='cuda:0', dtype=torch.int32) | |
| # _values | |
| tensor([[1., 2., 3., 4.], | |
| [5., 6., 7., 8.]], device='cuda:0') | |
| ########## torch.float32/torch.int32/size=(2, 3)+(3, 2)+() ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([[[0, 2, 4], | |
| [0, 3, 4], | |
| [0, 1, 4]], | |
| [[0, 1, 4], | |
| [0, 2, 4], | |
| [0, 3, 4]]]), | |
| row_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., 2., 3., 4.], | |
| [ 5., 6., 7., 8.], | |
| [ 9., 10., 11., 12.]], | |
| [[13., 14., 15., 16.], | |
| [17., 18., 19., 20.], | |
| [21., 22., 23., 24.]]]), device='cuda:0', | |
| size=(2, 3, 3, 2), nnz=4, layout=torch.sparse_csc) | |
| # _ccol_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) | |
| # _row_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., 2., 3., 4.], | |
| [ 5., 6., 7., 8.], | |
| [ 9., 10., 11., 12.]], | |
| [[13., 14., 15., 16.], | |
| [17., 18., 19., 20.], | |
| [21., 22., 23., 24.]]], device='cuda:0') | |
| ########## torch.float64/torch.int32/size=()+(3, 2)+() ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([0, 2, 4]), | |
| row_indices=tensor([0, 1, 0, 2]), | |
| values=tensor([1., 2., 3., 4.]), device='cuda:0', size=(3, 2), nnz=4, | |
| dtype=torch.float64, layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0, 2, 4], device='cuda:0', dtype=torch.int32) | |
| # _row_indices | |
| tensor([0, 1, 0, 2], device='cuda:0', dtype=torch.int32) | |
| # _values | |
| tensor([1., 2., 3., 4.], device='cuda:0', dtype=torch.float64) | |
| ########## torch.float64/torch.int32/size=()+(0, 0)+() ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([0]), | |
| row_indices=tensor([], size=(0,)), | |
| values=tensor([], size=(0,)), device='cuda:0', size=(0, 0), nnz=0, | |
| dtype=torch.float64, layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0], device='cuda:0', dtype=torch.int32) | |
| # _row_indices | |
| tensor([], device='cuda:0', dtype=torch.int32) | |
| # _values | |
| tensor([], device='cuda:0', dtype=torch.float64) | |
| ########## torch.float64/torch.int32/size=(2,)+(3, 2)+() ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([[0, 2, 4], | |
| [0, 3, 4]]), | |
| row_indices=tensor([[0, 1, 0, 1], | |
| [0, 1, 2, 0]]), | |
| values=tensor([[1., 2., 3., 4.], | |
| [5., 6., 7., 8.]]), device='cuda:0', size=(2, 3, 2), | |
| nnz=4, dtype=torch.float64, layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([[0, 2, 4], | |
| [0, 3, 4]], device='cuda:0', dtype=torch.int32) | |
| # _row_indices | |
| tensor([[0, 1, 0, 1], | |
| [0, 1, 2, 0]], device='cuda:0', dtype=torch.int32) | |
| # _values | |
| tensor([[1., 2., 3., 4.], | |
| [5., 6., 7., 8.]], device='cuda:0', dtype=torch.float64) | |
| ########## torch.float64/torch.int32/size=(2, 3)+(3, 2)+() ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([[[0, 2, 4], | |
| [0, 3, 4], | |
| [0, 1, 4]], | |
| [[0, 1, 4], | |
| [0, 2, 4], | |
| [0, 3, 4]]]), | |
| row_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., 2., 3., 4.], | |
| [ 5., 6., 7., 8.], | |
| [ 9., 10., 11., 12.]], | |
| [[13., 14., 15., 16.], | |
| [17., 18., 19., 20.], | |
| [21., 22., 23., 24.]]]), device='cuda:0', | |
| size=(2, 3, 3, 2), nnz=4, dtype=torch.float64, layout=torch.sparse_csc) | |
| # _ccol_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) | |
| # _row_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., 2., 3., 4.], | |
| [ 5., 6., 7., 8.], | |
| [ 9., 10., 11., 12.]], | |
| [[13., 14., 15., 16.], | |
| [17., 18., 19., 20.], | |
| [21., 22., 23., 24.]]], device='cuda:0', dtype=torch.float64) | |
| ########## torch.float32/torch.int64/size=()+(3, 2)+() ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([0, 2, 4]), | |
| row_indices=tensor([0, 1, 0, 2]), | |
| values=tensor([1., 2., 3., 4.]), device='cuda:0', size=(3, 2), nnz=4, | |
| layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0, 2, 4], device='cuda:0') | |
| # _row_indices | |
| tensor([0, 1, 0, 2], device='cuda:0') | |
| # _values | |
| tensor([1., 2., 3., 4.], device='cuda:0') | |
| ########## torch.float32/torch.int64/size=()+(0, 0)+() ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([0]), | |
| row_indices=tensor([], size=(0,)), | |
| values=tensor([], size=(0,)), device='cuda:0', size=(0, 0), nnz=0, | |
| layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0], device='cuda:0') | |
| # _row_indices | |
| tensor([], device='cuda:0', dtype=torch.int64) | |
| # _values | |
| tensor([], device='cuda:0') | |
| ########## torch.float32/torch.int64/size=(2,)+(3, 2)+() ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([[0, 2, 4], | |
| [0, 3, 4]]), | |
| row_indices=tensor([[0, 1, 0, 1], | |
| [0, 1, 2, 0]]), | |
| values=tensor([[1., 2., 3., 4.], | |
| [5., 6., 7., 8.]]), device='cuda:0', size=(2, 3, 2), | |
| nnz=4, layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([[0, 2, 4], | |
| [0, 3, 4]], device='cuda:0') | |
| # _row_indices | |
| tensor([[0, 1, 0, 1], | |
| [0, 1, 2, 0]], device='cuda:0') | |
| # _values | |
| tensor([[1., 2., 3., 4.], | |
| [5., 6., 7., 8.]], device='cuda:0') | |
| ########## torch.float32/torch.int64/size=(2, 3)+(3, 2)+() ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([[[0, 2, 4], | |
| [0, 3, 4], | |
| [0, 1, 4]], | |
| [[0, 1, 4], | |
| [0, 2, 4], | |
| [0, 3, 4]]]), | |
| row_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., 2., 3., 4.], | |
| [ 5., 6., 7., 8.], | |
| [ 9., 10., 11., 12.]], | |
| [[13., 14., 15., 16.], | |
| [17., 18., 19., 20.], | |
| [21., 22., 23., 24.]]]), device='cuda:0', | |
| size=(2, 3, 3, 2), nnz=4, layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([[[0, 2, 4], | |
| [0, 3, 4], | |
| [0, 1, 4]], | |
| [[0, 1, 4], | |
| [0, 2, 4], | |
| [0, 3, 4]]], device='cuda:0') | |
| # _row_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., 2., 3., 4.], | |
| [ 5., 6., 7., 8.], | |
| [ 9., 10., 11., 12.]], | |
| [[13., 14., 15., 16.], | |
| [17., 18., 19., 20.], | |
| [21., 22., 23., 24.]]], device='cuda:0') | |
| ########## torch.float64/torch.int64/size=()+(3, 2)+() ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([0, 2, 4]), | |
| row_indices=tensor([0, 1, 0, 2]), | |
| values=tensor([1., 2., 3., 4.]), device='cuda:0', size=(3, 2), nnz=4, | |
| dtype=torch.float64, layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0, 2, 4], device='cuda:0') | |
| # _row_indices | |
| tensor([0, 1, 0, 2], device='cuda:0') | |
| # _values | |
| tensor([1., 2., 3., 4.], device='cuda:0', dtype=torch.float64) | |
| ########## torch.float64/torch.int64/size=()+(0, 0)+() ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([0]), | |
| row_indices=tensor([], size=(0,)), | |
| values=tensor([], size=(0,)), device='cuda:0', size=(0, 0), nnz=0, | |
| dtype=torch.float64, layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0], device='cuda:0') | |
| # _row_indices | |
| tensor([], device='cuda:0', dtype=torch.int64) | |
| # _values | |
| tensor([], device='cuda:0', dtype=torch.float64) | |
| ########## torch.float64/torch.int64/size=(2,)+(3, 2)+() ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([[0, 2, 4], | |
| [0, 3, 4]]), | |
| row_indices=tensor([[0, 1, 0, 1], | |
| [0, 1, 2, 0]]), | |
| values=tensor([[1., 2., 3., 4.], | |
| [5., 6., 7., 8.]]), device='cuda:0', size=(2, 3, 2), | |
| nnz=4, dtype=torch.float64, layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([[0, 2, 4], | |
| [0, 3, 4]], device='cuda:0') | |
| # _row_indices | |
| tensor([[0, 1, 0, 1], | |
| [0, 1, 2, 0]], device='cuda:0') | |
| # _values | |
| tensor([[1., 2., 3., 4.], | |
| [5., 6., 7., 8.]], device='cuda:0', dtype=torch.float64) | |
| ########## torch.float64/torch.int64/size=(2, 3)+(3, 2)+() ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([[[0, 2, 4], | |
| [0, 3, 4], | |
| [0, 1, 4]], | |
| [[0, 1, 4], | |
| [0, 2, 4], | |
| [0, 3, 4]]]), | |
| row_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., 2., 3., 4.], | |
| [ 5., 6., 7., 8.], | |
| [ 9., 10., 11., 12.]], | |
| [[13., 14., 15., 16.], | |
| [17., 18., 19., 20.], | |
| [21., 22., 23., 24.]]]), device='cuda:0', | |
| size=(2, 3, 3, 2), nnz=4, dtype=torch.float64, layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([[[0, 2, 4], | |
| [0, 3, 4], | |
| [0, 1, 4]], | |
| [[0, 1, 4], | |
| [0, 2, 4], | |
| [0, 3, 4]]], device='cuda:0') | |
| # _row_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., 2., 3., 4.], | |
| [ 5., 6., 7., 8.], | |
| [ 9., 10., 11., 12.]], | |
| [[13., 14., 15., 16.], | |
| [17., 18., 19., 20.], | |
| [21., 22., 23., 24.]]], device='cuda:0', dtype=torch.float64) | |
| ########## torch.float32/torch.int32/size=()+(3, 2)+(2,) ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([0, 2, 4]), | |
| row_indices=tensor([0, 1, 0, 2]), | |
| values=tensor([[1., 2.], | |
| [2., 3.], | |
| [3., 4.], | |
| [4., 5.]]), device='cuda:0', size=(3, 2, 2), nnz=4, | |
| layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0, 2, 4], device='cuda:0', dtype=torch.int32) | |
| # _row_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=()+(3, 2)+(4, 2) ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([0, 2, 4]), | |
| row_indices=tensor([0, 1, 0, 2]), | |
| values=tensor([[[ 1., 11.], | |
| [ 2., 12.], | |
| [ 3., 13.], | |
| [ 4., 14.]], | |
| [[ 2., 12.], | |
| [ 3., 13.], | |
| [ 4., 14.], | |
| [ 5., 15.]], | |
| [[ 3., 13.], | |
| [ 4., 14.], | |
| [ 5., 15.], | |
| [ 6., 16.]], | |
| [[ 4., 14.], | |
| [ 5., 15.], | |
| [ 6., 16.], | |
| [ 7., 17.]]]), device='cuda:0', size=(3, 2, 4, 2), | |
| nnz=4, layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0, 2, 4], device='cuda:0', dtype=torch.int32) | |
| # _row_indices | |
| tensor([0, 1, 0, 2], device='cuda:0', dtype=torch.int32) | |
| # _values | |
| tensor([[[ 1., 11.], | |
| [ 2., 12.], | |
| [ 3., 13.], | |
| [ 4., 14.]], | |
| [[ 2., 12.], | |
| [ 3., 13.], | |
| [ 4., 14.], | |
| [ 5., 15.]], | |
| [[ 3., 13.], | |
| [ 4., 14.], | |
| [ 5., 15.], | |
| [ 6., 16.]], | |
| [[ 4., 14.], | |
| [ 5., 15.], | |
| [ 6., 16.], | |
| [ 7., 17.]]], device='cuda:0') | |
| ########## torch.float32/torch.int32/size=(2, 3)+(3, 2)+(2, 1) ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([[[0, 2, 4], | |
| [0, 3, 4], | |
| [0, 1, 4]], | |
| [[0, 1, 4], | |
| [0, 2, 4], | |
| [0, 3, 4]]]), | |
| row_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.], | |
| [ 2.]], | |
| [[ 2.], | |
| [ 3.]], | |
| [[ 3.], | |
| [ 4.]], | |
| [[ 4.], | |
| [ 5.]]], | |
| [[[ 5.], | |
| [ 6.]], | |
| [[ 6.], | |
| [ 7.]], | |
| [[ 7.], | |
| [ 8.]], | |
| [[ 8.], | |
| [ 9.]]], | |
| [[[ 9.], | |
| [10.]], | |
| [[10.], | |
| [11.]], | |
| [[11.], | |
| [12.]], | |
| [[12.], | |
| [13.]]]], | |
| [[[[13.], | |
| [14.]], | |
| [[14.], | |
| [15.]], | |
| [[15.], | |
| [16.]], | |
| [[16.], | |
| [17.]]], | |
| [[[17.], | |
| [18.]], | |
| [[18.], | |
| [19.]], | |
| [[19.], | |
| [20.]], | |
| [[20.], | |
| [21.]]], | |
| [[[21.], | |
| [22.]], | |
| [[22.], | |
| [23.]], | |
| [[23.], | |
| [24.]], | |
| [[24.], | |
| [25.]]]]]), device='cuda:0', size=(2, 3, 3, 2, 2, 1), | |
| nnz=4, layout=torch.sparse_csc) | |
| # _ccol_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) | |
| # _row_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.], | |
| [ 2.]], | |
| [[ 2.], | |
| [ 3.]], | |
| [[ 3.], | |
| [ 4.]], | |
| [[ 4.], | |
| [ 5.]]], | |
| [[[ 5.], | |
| [ 6.]], | |
| [[ 6.], | |
| [ 7.]], | |
| [[ 7.], | |
| [ 8.]], | |
| [[ 8.], | |
| [ 9.]]], | |
| [[[ 9.], | |
| [10.]], | |
| [[10.], | |
| [11.]], | |
| [[11.], | |
| [12.]], | |
| [[12.], | |
| [13.]]]], | |
| [[[[13.], | |
| [14.]], | |
| [[14.], | |
| [15.]], | |
| [[15.], | |
| [16.]], | |
| [[16.], | |
| [17.]]], | |
| [[[17.], | |
| [18.]], | |
| [[18.], | |
| [19.]], | |
| [[19.], | |
| [20.]], | |
| [[20.], | |
| [21.]]], | |
| [[[21.], | |
| [22.]], | |
| [[22.], | |
| [23.]], | |
| [[23.], | |
| [24.]], | |
| [[24.], | |
| [25.]]]]], device='cuda:0') | |
| ########## torch.float64/torch.int32/size=()+(3, 2)+(2,) ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([0, 2, 4]), | |
| row_indices=tensor([0, 1, 0, 2]), | |
| values=tensor([[1., 2.], | |
| [2., 3.], | |
| [3., 4.], | |
| [4., 5.]]), device='cuda:0', size=(3, 2, 2), nnz=4, | |
| dtype=torch.float64, layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0, 2, 4], device='cuda:0', dtype=torch.int32) | |
| # _row_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=()+(3, 2)+(4, 2) ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([0, 2, 4]), | |
| row_indices=tensor([0, 1, 0, 2]), | |
| values=tensor([[[ 1., 11.], | |
| [ 2., 12.], | |
| [ 3., 13.], | |
| [ 4., 14.]], | |
| [[ 2., 12.], | |
| [ 3., 13.], | |
| [ 4., 14.], | |
| [ 5., 15.]], | |
| [[ 3., 13.], | |
| [ 4., 14.], | |
| [ 5., 15.], | |
| [ 6., 16.]], | |
| [[ 4., 14.], | |
| [ 5., 15.], | |
| [ 6., 16.], | |
| [ 7., 17.]]]), device='cuda:0', size=(3, 2, 4, 2), | |
| nnz=4, dtype=torch.float64, layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0, 2, 4], device='cuda:0', dtype=torch.int32) | |
| # _row_indices | |
| tensor([0, 1, 0, 2], device='cuda:0', dtype=torch.int32) | |
| # _values | |
| tensor([[[ 1., 11.], | |
| [ 2., 12.], | |
| [ 3., 13.], | |
| [ 4., 14.]], | |
| [[ 2., 12.], | |
| [ 3., 13.], | |
| [ 4., 14.], | |
| [ 5., 15.]], | |
| [[ 3., 13.], | |
| [ 4., 14.], | |
| [ 5., 15.], | |
| [ 6., 16.]], | |
| [[ 4., 14.], | |
| [ 5., 15.], | |
| [ 6., 16.], | |
| [ 7., 17.]]], device='cuda:0', dtype=torch.float64) | |
| ########## torch.float64/torch.int32/size=(2, 3)+(3, 2)+(2, 1) ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([[[0, 2, 4], | |
| [0, 3, 4], | |
| [0, 1, 4]], | |
| [[0, 1, 4], | |
| [0, 2, 4], | |
| [0, 3, 4]]]), | |
| row_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.], | |
| [ 2.]], | |
| [[ 2.], | |
| [ 3.]], | |
| [[ 3.], | |
| [ 4.]], | |
| [[ 4.], | |
| [ 5.]]], | |
| [[[ 5.], | |
| [ 6.]], | |
| [[ 6.], | |
| [ 7.]], | |
| [[ 7.], | |
| [ 8.]], | |
| [[ 8.], | |
| [ 9.]]], | |
| [[[ 9.], | |
| [10.]], | |
| [[10.], | |
| [11.]], | |
| [[11.], | |
| [12.]], | |
| [[12.], | |
| [13.]]]], | |
| [[[[13.], | |
| [14.]], | |
| [[14.], | |
| [15.]], | |
| [[15.], | |
| [16.]], | |
| [[16.], | |
| [17.]]], | |
| [[[17.], | |
| [18.]], | |
| [[18.], | |
| [19.]], | |
| [[19.], | |
| [20.]], | |
| [[20.], | |
| [21.]]], | |
| [[[21.], | |
| [22.]], | |
| [[22.], | |
| [23.]], | |
| [[23.], | |
| [24.]], | |
| [[24.], | |
| [25.]]]]]), device='cuda:0', size=(2, 3, 3, 2, 2, 1), | |
| nnz=4, dtype=torch.float64, layout=torch.sparse_csc) | |
| # _ccol_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) | |
| # _row_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.], | |
| [ 2.]], | |
| [[ 2.], | |
| [ 3.]], | |
| [[ 3.], | |
| [ 4.]], | |
| [[ 4.], | |
| [ 5.]]], | |
| [[[ 5.], | |
| [ 6.]], | |
| [[ 6.], | |
| [ 7.]], | |
| [[ 7.], | |
| [ 8.]], | |
| [[ 8.], | |
| [ 9.]]], | |
| [[[ 9.], | |
| [10.]], | |
| [[10.], | |
| [11.]], | |
| [[11.], | |
| [12.]], | |
| [[12.], | |
| [13.]]]], | |
| [[[[13.], | |
| [14.]], | |
| [[14.], | |
| [15.]], | |
| [[15.], | |
| [16.]], | |
| [[16.], | |
| [17.]]], | |
| [[[17.], | |
| [18.]], | |
| [[18.], | |
| [19.]], | |
| [[19.], | |
| [20.]], | |
| [[20.], | |
| [21.]]], | |
| [[[21.], | |
| [22.]], | |
| [[22.], | |
| [23.]], | |
| [[23.], | |
| [24.]], | |
| [[24.], | |
| [25.]]]]], device='cuda:0', dtype=torch.float64) | |
| ########## torch.float32/torch.int64/size=()+(3, 2)+(2,) ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([0, 2, 4]), | |
| row_indices=tensor([0, 1, 0, 2]), | |
| values=tensor([[1., 2.], | |
| [2., 3.], | |
| [3., 4.], | |
| [4., 5.]]), device='cuda:0', size=(3, 2, 2), nnz=4, | |
| layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0, 2, 4], device='cuda:0') | |
| # _row_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=()+(3, 2)+(4, 2) ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([0, 2, 4]), | |
| row_indices=tensor([0, 1, 0, 2]), | |
| values=tensor([[[ 1., 11.], | |
| [ 2., 12.], | |
| [ 3., 13.], | |
| [ 4., 14.]], | |
| [[ 2., 12.], | |
| [ 3., 13.], | |
| [ 4., 14.], | |
| [ 5., 15.]], | |
| [[ 3., 13.], | |
| [ 4., 14.], | |
| [ 5., 15.], | |
| [ 6., 16.]], | |
| [[ 4., 14.], | |
| [ 5., 15.], | |
| [ 6., 16.], | |
| [ 7., 17.]]]), device='cuda:0', size=(3, 2, 4, 2), | |
| nnz=4, layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0, 2, 4], device='cuda:0') | |
| # _row_indices | |
| tensor([0, 1, 0, 2], device='cuda:0') | |
| # _values | |
| tensor([[[ 1., 11.], | |
| [ 2., 12.], | |
| [ 3., 13.], | |
| [ 4., 14.]], | |
| [[ 2., 12.], | |
| [ 3., 13.], | |
| [ 4., 14.], | |
| [ 5., 15.]], | |
| [[ 3., 13.], | |
| [ 4., 14.], | |
| [ 5., 15.], | |
| [ 6., 16.]], | |
| [[ 4., 14.], | |
| [ 5., 15.], | |
| [ 6., 16.], | |
| [ 7., 17.]]], device='cuda:0') | |
| ########## torch.float32/torch.int64/size=(2, 3)+(3, 2)+(2, 1) ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([[[0, 2, 4], | |
| [0, 3, 4], | |
| [0, 1, 4]], | |
| [[0, 1, 4], | |
| [0, 2, 4], | |
| [0, 3, 4]]]), | |
| row_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.], | |
| [ 2.]], | |
| [[ 2.], | |
| [ 3.]], | |
| [[ 3.], | |
| [ 4.]], | |
| [[ 4.], | |
| [ 5.]]], | |
| [[[ 5.], | |
| [ 6.]], | |
| [[ 6.], | |
| [ 7.]], | |
| [[ 7.], | |
| [ 8.]], | |
| [[ 8.], | |
| [ 9.]]], | |
| [[[ 9.], | |
| [10.]], | |
| [[10.], | |
| [11.]], | |
| [[11.], | |
| [12.]], | |
| [[12.], | |
| [13.]]]], | |
| [[[[13.], | |
| [14.]], | |
| [[14.], | |
| [15.]], | |
| [[15.], | |
| [16.]], | |
| [[16.], | |
| [17.]]], | |
| [[[17.], | |
| [18.]], | |
| [[18.], | |
| [19.]], | |
| [[19.], | |
| [20.]], | |
| [[20.], | |
| [21.]]], | |
| [[[21.], | |
| [22.]], | |
| [[22.], | |
| [23.]], | |
| [[23.], | |
| [24.]], | |
| [[24.], | |
| [25.]]]]]), device='cuda:0', size=(2, 3, 3, 2, 2, 1), | |
| nnz=4, layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([[[0, 2, 4], | |
| [0, 3, 4], | |
| [0, 1, 4]], | |
| [[0, 1, 4], | |
| [0, 2, 4], | |
| [0, 3, 4]]], device='cuda:0') | |
| # _row_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.], | |
| [ 2.]], | |
| [[ 2.], | |
| [ 3.]], | |
| [[ 3.], | |
| [ 4.]], | |
| [[ 4.], | |
| [ 5.]]], | |
| [[[ 5.], | |
| [ 6.]], | |
| [[ 6.], | |
| [ 7.]], | |
| [[ 7.], | |
| [ 8.]], | |
| [[ 8.], | |
| [ 9.]]], | |
| [[[ 9.], | |
| [10.]], | |
| [[10.], | |
| [11.]], | |
| [[11.], | |
| [12.]], | |
| [[12.], | |
| [13.]]]], | |
| [[[[13.], | |
| [14.]], | |
| [[14.], | |
| [15.]], | |
| [[15.], | |
| [16.]], | |
| [[16.], | |
| [17.]]], | |
| [[[17.], | |
| [18.]], | |
| [[18.], | |
| [19.]], | |
| [[19.], | |
| [20.]], | |
| [[20.], | |
| [21.]]], | |
| [[[21.], | |
| [22.]], | |
| [[22.], | |
| [23.]], | |
| [[23.], | |
| [24.]], | |
| [[24.], | |
| [25.]]]]], device='cuda:0') | |
| ########## torch.float64/torch.int64/size=()+(3, 2)+(2,) ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([0, 2, 4]), | |
| row_indices=tensor([0, 1, 0, 2]), | |
| values=tensor([[1., 2.], | |
| [2., 3.], | |
| [3., 4.], | |
| [4., 5.]]), device='cuda:0', size=(3, 2, 2), nnz=4, | |
| dtype=torch.float64, layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0, 2, 4], device='cuda:0') | |
| # _row_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=()+(3, 2)+(4, 2) ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([0, 2, 4]), | |
| row_indices=tensor([0, 1, 0, 2]), | |
| values=tensor([[[ 1., 11.], | |
| [ 2., 12.], | |
| [ 3., 13.], | |
| [ 4., 14.]], | |
| [[ 2., 12.], | |
| [ 3., 13.], | |
| [ 4., 14.], | |
| [ 5., 15.]], | |
| [[ 3., 13.], | |
| [ 4., 14.], | |
| [ 5., 15.], | |
| [ 6., 16.]], | |
| [[ 4., 14.], | |
| [ 5., 15.], | |
| [ 6., 16.], | |
| [ 7., 17.]]]), device='cuda:0', size=(3, 2, 4, 2), | |
| nnz=4, dtype=torch.float64, layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0, 2, 4], device='cuda:0') | |
| # _row_indices | |
| tensor([0, 1, 0, 2], device='cuda:0') | |
| # _values | |
| tensor([[[ 1., 11.], | |
| [ 2., 12.], | |
| [ 3., 13.], | |
| [ 4., 14.]], | |
| [[ 2., 12.], | |
| [ 3., 13.], | |
| [ 4., 14.], | |
| [ 5., 15.]], | |
| [[ 3., 13.], | |
| [ 4., 14.], | |
| [ 5., 15.], | |
| [ 6., 16.]], | |
| [[ 4., 14.], | |
| [ 5., 15.], | |
| [ 6., 16.], | |
| [ 7., 17.]]], device='cuda:0', dtype=torch.float64) | |
| ########## torch.float64/torch.int64/size=(2, 3)+(3, 2)+(2, 1) ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([[[0, 2, 4], | |
| [0, 3, 4], | |
| [0, 1, 4]], | |
| [[0, 1, 4], | |
| [0, 2, 4], | |
| [0, 3, 4]]]), | |
| row_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.], | |
| [ 2.]], | |
| [[ 2.], | |
| [ 3.]], | |
| [[ 3.], | |
| [ 4.]], | |
| [[ 4.], | |
| [ 5.]]], | |
| [[[ 5.], | |
| [ 6.]], | |
| [[ 6.], | |
| [ 7.]], | |
| [[ 7.], | |
| [ 8.]], | |
| [[ 8.], | |
| [ 9.]]], | |
| [[[ 9.], | |
| [10.]], | |
| [[10.], | |
| [11.]], | |
| [[11.], | |
| [12.]], | |
| [[12.], | |
| [13.]]]], | |
| [[[[13.], | |
| [14.]], | |
| [[14.], | |
| [15.]], | |
| [[15.], | |
| [16.]], | |
| [[16.], | |
| [17.]]], | |
| [[[17.], | |
| [18.]], | |
| [[18.], | |
| [19.]], | |
| [[19.], | |
| [20.]], | |
| [[20.], | |
| [21.]]], | |
| [[[21.], | |
| [22.]], | |
| [[22.], | |
| [23.]], | |
| [[23.], | |
| [24.]], | |
| [[24.], | |
| [25.]]]]]), device='cuda:0', size=(2, 3, 3, 2, 2, 1), | |
| nnz=4, dtype=torch.float64, layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([[[0, 2, 4], | |
| [0, 3, 4], | |
| [0, 1, 4]], | |
| [[0, 1, 4], | |
| [0, 2, 4], | |
| [0, 3, 4]]], device='cuda:0') | |
| # _row_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.], | |
| [ 2.]], | |
| [[ 2.], | |
| [ 3.]], | |
| [[ 3.], | |
| [ 4.]], | |
| [[ 4.], | |
| [ 5.]]], | |
| [[[ 5.], | |
| [ 6.]], | |
| [[ 6.], | |
| [ 7.]], | |
| [[ 7.], | |
| [ 8.]], | |
| [[ 8.], | |
| [ 9.]]], | |
| [[[ 9.], | |
| [10.]], | |
| [[10.], | |
| [11.]], | |
| [[11.], | |
| [12.]], | |
| [[12.], | |
| [13.]]]], | |
| [[[[13.], | |
| [14.]], | |
| [[14.], | |
| [15.]], | |
| [[15.], | |
| [16.]], | |
| [[16.], | |
| [17.]]], | |
| [[[17.], | |
| [18.]], | |
| [[18.], | |
| [19.]], | |
| [[19.], | |
| [20.]], | |
| [[20.], | |
| [21.]]], | |
| [[[21.], | |
| [22.]], | |
| [[22.], | |
| [23.]], | |
| [[23.], | |
| [24.]], | |
| [[24.], | |
| [25.]]]]], device='cuda:0', dtype=torch.float64) | |