| ########## 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.]), size=(3, 2), nnz=4, | |
| layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0, 2, 4], dtype=torch.int32) | |
| # _row_indices | |
| tensor([0, 1, 0, 2], dtype=torch.int32) | |
| # _values | |
| tensor([1., 2., 3., 4.]) | |
| ########## torch.float32/torch.int32/size=()+(0, 0)+() ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([0]), | |
| row_indices=tensor([], size=(0,)), | |
| values=tensor([], size=(0,)), size=(0, 0), nnz=0, | |
| layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0], dtype=torch.int32) | |
| # _row_indices | |
| tensor([], dtype=torch.int32) | |
| # _values | |
| tensor([]) | |
| ########## 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.]]), size=(2, 3, 2), nnz=4, | |
| layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([[0, 2, 4], | |
| [0, 3, 4]], dtype=torch.int32) | |
| # _row_indices | |
| tensor([[0, 1, 0, 1], | |
| [0, 1, 2, 0]], dtype=torch.int32) | |
| # _values | |
| tensor([[1., 2., 3., 4.], | |
| [5., 6., 7., 8.]]) | |
| ########## 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.]]]), 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]]], 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]]], 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.]]]) | |
| ########## 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.]), size=(3, 2), nnz=4, | |
| dtype=torch.float64, layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0, 2, 4], dtype=torch.int32) | |
| # _row_indices | |
| tensor([0, 1, 0, 2], dtype=torch.int32) | |
| # _values | |
| tensor([1., 2., 3., 4.], 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,)), size=(0, 0), nnz=0, dtype=torch.float64, | |
| layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0], dtype=torch.int32) | |
| # _row_indices | |
| tensor([], dtype=torch.int32) | |
| # _values | |
| tensor([], 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.]]), size=(2, 3, 2), nnz=4, | |
| dtype=torch.float64, layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([[0, 2, 4], | |
| [0, 3, 4]], dtype=torch.int32) | |
| # _row_indices | |
| tensor([[0, 1, 0, 1], | |
| [0, 1, 2, 0]], dtype=torch.int32) | |
| # _values | |
| tensor([[1., 2., 3., 4.], | |
| [5., 6., 7., 8.]], 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.]]]), 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]]], 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]]], 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.]]], 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.]), size=(3, 2), nnz=4, | |
| layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0, 2, 4]) | |
| # _row_indices | |
| tensor([0, 1, 0, 2]) | |
| # _values | |
| tensor([1., 2., 3., 4.]) | |
| ########## torch.float32/torch.int64/size=()+(0, 0)+() ########## | |
| # sparse tensor | |
| tensor(ccol_indices=tensor([0]), | |
| row_indices=tensor([], size=(0,)), | |
| values=tensor([], size=(0,)), size=(0, 0), nnz=0, | |
| layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0]) | |
| # _row_indices | |
| tensor([], dtype=torch.int64) | |
| # _values | |
| tensor([]) | |
| ########## 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.]]), size=(2, 3, 2), nnz=4, | |
| layout=torch.sparse_csc) | |
| # _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.]]) | |
| ########## 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.]]]), 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]]]) | |
| # _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.]]]) | |
| ########## 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.]), size=(3, 2), nnz=4, | |
| dtype=torch.float64, layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0, 2, 4]) | |
| # _row_indices | |
| tensor([0, 1, 0, 2]) | |
| # _values | |
| tensor([1., 2., 3., 4.], 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,)), size=(0, 0), nnz=0, dtype=torch.float64, | |
| layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0]) | |
| # _row_indices | |
| tensor([], dtype=torch.int64) | |
| # _values | |
| tensor([], 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.]]), size=(2, 3, 2), nnz=4, | |
| dtype=torch.float64, layout=torch.sparse_csc) | |
| # _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.]], 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.]]]), 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]]]) | |
| # _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.]]], 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.]]), size=(3, 2, 2), nnz=4, | |
| layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0, 2, 4], dtype=torch.int32) | |
| # _row_indices | |
| tensor([0, 1, 0, 2], dtype=torch.int32) | |
| # _values | |
| tensor([[1., 2.], | |
| [2., 3.], | |
| [3., 4.], | |
| [4., 5.]]) | |
| ########## 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.]]]), size=(3, 2, 4, 2), nnz=4, | |
| layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0, 2, 4], dtype=torch.int32) | |
| # _row_indices | |
| tensor([0, 1, 0, 2], 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.]]]) | |
| ########## 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.]]]]]), 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]]], 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]]], 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.]]]]]) | |
| ########## 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.]]), size=(3, 2, 2), nnz=4, dtype=torch.float64, | |
| layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0, 2, 4], dtype=torch.int32) | |
| # _row_indices | |
| tensor([0, 1, 0, 2], dtype=torch.int32) | |
| # _values | |
| tensor([[1., 2.], | |
| [2., 3.], | |
| [3., 4.], | |
| [4., 5.]], 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.]]]), size=(3, 2, 4, 2), nnz=4, | |
| dtype=torch.float64, layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0, 2, 4], dtype=torch.int32) | |
| # _row_indices | |
| tensor([0, 1, 0, 2], 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.]]], 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.]]]]]), 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]]], 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]]], 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.]]]]], 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.]]), size=(3, 2, 2), nnz=4, | |
| layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0, 2, 4]) | |
| # _row_indices | |
| tensor([0, 1, 0, 2]) | |
| # _values | |
| tensor([[1., 2.], | |
| [2., 3.], | |
| [3., 4.], | |
| [4., 5.]]) | |
| ########## 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.]]]), size=(3, 2, 4, 2), nnz=4, | |
| layout=torch.sparse_csc) | |
| # _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.]]]) | |
| ########## 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.]]]]]), 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]]]) | |
| # _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.]]]]]) | |
| ########## 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.]]), size=(3, 2, 2), nnz=4, dtype=torch.float64, | |
| layout=torch.sparse_csc) | |
| # _ccol_indices | |
| tensor([0, 2, 4]) | |
| # _row_indices | |
| tensor([0, 1, 0, 2]) | |
| # _values | |
| tensor([[1., 2.], | |
| [2., 3.], | |
| [3., 4.], | |
| [4., 5.]], 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.]]]), size=(3, 2, 4, 2), nnz=4, | |
| dtype=torch.float64, layout=torch.sparse_csc) | |
| # _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.]]], 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.]]]]]), 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]]]) | |
| # _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.]]]]], dtype=torch.float64) | |