########## torch.float32/torch.int32/size=(2, 3)+(2, 3)+() ########## # sparse tensor tensor(crow_indices=tensor([[[0, 3], [0, 3], [0, 3]], [[0, 3], [0, 3], [0, 3]]]), col_indices=tensor([[[0, 1, 2], [0, 1, 2], [0, 1, 2]], [[0, 1, 2], [0, 1, 2], [0, 1, 2]]]), values=tensor([[[[[1.], [3.]], [[2.], [0.]], [[0.], [4.]]], [[[1.], [4.]], [[2.], [0.]], [[3.], [0.]]], [[[1.], [2.]], [[0.], [3.]], [[0.], [4.]]]], [[[[0.], [2.]], [[1.], [3.]], [[0.], [4.]]], [[[1.], [3.]], [[0.], [4.]], [[2.], [0.]]], [[[1.], [0.]], [[2.], [4.]], [[3.], [0.]]]]]), size=(2, 3, 2, 3), nnz=3, layout=torch.sparse_bsr) # _crow_indices tensor([[[0, 3], [0, 3], [0, 3]], [[0, 3], [0, 3], [0, 3]]], dtype=torch.int32) # _col_indices tensor([[[0, 1, 2], [0, 1, 2], [0, 1, 2]], [[0, 1, 2], [0, 1, 2], [0, 1, 2]]], dtype=torch.int32) # _values tensor([[[[[1.], [3.]], [[2.], [0.]], [[0.], [4.]]], [[[1.], [4.]], [[2.], [0.]], [[3.], [0.]]], [[[1.], [2.]], [[0.], [3.]], [[0.], [4.]]]], [[[[0.], [2.]], [[1.], [3.]], [[0.], [4.]]], [[[1.], [3.]], [[0.], [4.]], [[2.], [0.]]], [[[1.], [0.]], [[2.], [4.]], [[3.], [0.]]]]]) ########## torch.float32/torch.int32/size=()+(8, 6)+() ########## # sparse tensor tensor(crow_indices=tensor([0, 2, 3, 5, 7]), col_indices=tensor([0, 1, 0, 0, 1, 0, 1]), values=tensor([[[ 0., 1., 0.], [ 0., 4., 0.]], [[ 2., 0., 3.], [ 0., 5., 0.]], [[ 6., 7., 8.], [ 0., 0., 0.]], [[ 0., 9., 0.], [13., 0., 14.]], [[10., 11., 12.], [15., 16., 17.]], [[ 0., 0., 0.], [20., 21., 22.]], [[ 0., 18., 19.], [ 0., 23., 24.]]]), size=(8, 6), nnz=7, layout=torch.sparse_bsr) # _crow_indices tensor([0, 2, 3, 5, 7], dtype=torch.int32) # _col_indices tensor([0, 1, 0, 0, 1, 0, 1], dtype=torch.int32) # _values tensor([[[ 0., 1., 0.], [ 0., 4., 0.]], [[ 2., 0., 3.], [ 0., 5., 0.]], [[ 6., 7., 8.], [ 0., 0., 0.]], [[ 0., 9., 0.], [13., 0., 14.]], [[10., 11., 12.], [15., 16., 17.]], [[ 0., 0., 0.], [20., 21., 22.]], [[ 0., 18., 19.], [ 0., 23., 24.]]]) ########## torch.float64/torch.int32/size=(2, 3)+(2, 3)+() ########## # sparse tensor tensor(crow_indices=tensor([[[0, 3], [0, 3], [0, 3]], [[0, 3], [0, 3], [0, 3]]]), col_indices=tensor([[[0, 1, 2], [0, 1, 2], [0, 1, 2]], [[0, 1, 2], [0, 1, 2], [0, 1, 2]]]), values=tensor([[[[[1.], [3.]], [[2.], [0.]], [[0.], [4.]]], [[[1.], [4.]], [[2.], [0.]], [[3.], [0.]]], [[[1.], [2.]], [[0.], [3.]], [[0.], [4.]]]], [[[[0.], [2.]], [[1.], [3.]], [[0.], [4.]]], [[[1.], [3.]], [[0.], [4.]], [[2.], [0.]]], [[[1.], [0.]], [[2.], [4.]], [[3.], [0.]]]]]), size=(2, 3, 2, 3), nnz=3, dtype=torch.float64, layout=torch.sparse_bsr) # _crow_indices tensor([[[0, 3], [0, 3], [0, 3]], [[0, 3], [0, 3], [0, 3]]], dtype=torch.int32) # _col_indices tensor([[[0, 1, 2], [0, 1, 2], [0, 1, 2]], [[0, 1, 2], [0, 1, 2], [0, 1, 2]]], dtype=torch.int32) # _values tensor([[[[[1.], [3.]], [[2.], [0.]], [[0.], [4.]]], [[[1.], [4.]], [[2.], [0.]], [[3.], [0.]]], [[[1.], [2.]], [[0.], [3.]], [[0.], [4.]]]], [[[[0.], [2.]], [[1.], [3.]], [[0.], [4.]]], [[[1.], [3.]], [[0.], [4.]], [[2.], [0.]]], [[[1.], [0.]], [[2.], [4.]], [[3.], [0.]]]]], dtype=torch.float64) ########## torch.float64/torch.int32/size=()+(8, 6)+() ########## # sparse tensor tensor(crow_indices=tensor([0, 2, 3, 5, 7]), col_indices=tensor([0, 1, 0, 0, 1, 0, 1]), values=tensor([[[ 0., 1., 0.], [ 0., 4., 0.]], [[ 2., 0., 3.], [ 0., 5., 0.]], [[ 6., 7., 8.], [ 0., 0., 0.]], [[ 0., 9., 0.], [13., 0., 14.]], [[10., 11., 12.], [15., 16., 17.]], [[ 0., 0., 0.], [20., 21., 22.]], [[ 0., 18., 19.], [ 0., 23., 24.]]]), size=(8, 6), nnz=7, dtype=torch.float64, layout=torch.sparse_bsr) # _crow_indices tensor([0, 2, 3, 5, 7], dtype=torch.int32) # _col_indices tensor([0, 1, 0, 0, 1, 0, 1], dtype=torch.int32) # _values tensor([[[ 0., 1., 0.], [ 0., 4., 0.]], [[ 2., 0., 3.], [ 0., 5., 0.]], [[ 6., 7., 8.], [ 0., 0., 0.]], [[ 0., 9., 0.], [13., 0., 14.]], [[10., 11., 12.], [15., 16., 17.]], [[ 0., 0., 0.], [20., 21., 22.]], [[ 0., 18., 19.], [ 0., 23., 24.]]], dtype=torch.float64) ########## torch.float32/torch.int64/size=(2, 3)+(2, 3)+() ########## # sparse tensor tensor(crow_indices=tensor([[[0, 3], [0, 3], [0, 3]], [[0, 3], [0, 3], [0, 3]]]), col_indices=tensor([[[0, 1, 2], [0, 1, 2], [0, 1, 2]], [[0, 1, 2], [0, 1, 2], [0, 1, 2]]]), values=tensor([[[[[1.], [3.]], [[2.], [0.]], [[0.], [4.]]], [[[1.], [4.]], [[2.], [0.]], [[3.], [0.]]], [[[1.], [2.]], [[0.], [3.]], [[0.], [4.]]]], [[[[0.], [2.]], [[1.], [3.]], [[0.], [4.]]], [[[1.], [3.]], [[0.], [4.]], [[2.], [0.]]], [[[1.], [0.]], [[2.], [4.]], [[3.], [0.]]]]]), size=(2, 3, 2, 3), nnz=3, layout=torch.sparse_bsr) # _crow_indices tensor([[[0, 3], [0, 3], [0, 3]], [[0, 3], [0, 3], [0, 3]]]) # _col_indices tensor([[[0, 1, 2], [0, 1, 2], [0, 1, 2]], [[0, 1, 2], [0, 1, 2], [0, 1, 2]]]) # _values tensor([[[[[1.], [3.]], [[2.], [0.]], [[0.], [4.]]], [[[1.], [4.]], [[2.], [0.]], [[3.], [0.]]], [[[1.], [2.]], [[0.], [3.]], [[0.], [4.]]]], [[[[0.], [2.]], [[1.], [3.]], [[0.], [4.]]], [[[1.], [3.]], [[0.], [4.]], [[2.], [0.]]], [[[1.], [0.]], [[2.], [4.]], [[3.], [0.]]]]]) ########## torch.float32/torch.int64/size=()+(8, 6)+() ########## # sparse tensor tensor(crow_indices=tensor([0, 2, 3, 5, 7]), col_indices=tensor([0, 1, 0, 0, 1, 0, 1]), values=tensor([[[ 0., 1., 0.], [ 0., 4., 0.]], [[ 2., 0., 3.], [ 0., 5., 0.]], [[ 6., 7., 8.], [ 0., 0., 0.]], [[ 0., 9., 0.], [13., 0., 14.]], [[10., 11., 12.], [15., 16., 17.]], [[ 0., 0., 0.], [20., 21., 22.]], [[ 0., 18., 19.], [ 0., 23., 24.]]]), size=(8, 6), nnz=7, layout=torch.sparse_bsr) # _crow_indices tensor([0, 2, 3, 5, 7]) # _col_indices tensor([0, 1, 0, 0, 1, 0, 1]) # _values tensor([[[ 0., 1., 0.], [ 0., 4., 0.]], [[ 2., 0., 3.], [ 0., 5., 0.]], [[ 6., 7., 8.], [ 0., 0., 0.]], [[ 0., 9., 0.], [13., 0., 14.]], [[10., 11., 12.], [15., 16., 17.]], [[ 0., 0., 0.], [20., 21., 22.]], [[ 0., 18., 19.], [ 0., 23., 24.]]]) ########## torch.float64/torch.int64/size=(2, 3)+(2, 3)+() ########## # sparse tensor tensor(crow_indices=tensor([[[0, 3], [0, 3], [0, 3]], [[0, 3], [0, 3], [0, 3]]]), col_indices=tensor([[[0, 1, 2], [0, 1, 2], [0, 1, 2]], [[0, 1, 2], [0, 1, 2], [0, 1, 2]]]), values=tensor([[[[[1.], [3.]], [[2.], [0.]], [[0.], [4.]]], [[[1.], [4.]], [[2.], [0.]], [[3.], [0.]]], [[[1.], [2.]], [[0.], [3.]], [[0.], [4.]]]], [[[[0.], [2.]], [[1.], [3.]], [[0.], [4.]]], [[[1.], [3.]], [[0.], [4.]], [[2.], [0.]]], [[[1.], [0.]], [[2.], [4.]], [[3.], [0.]]]]]), size=(2, 3, 2, 3), nnz=3, dtype=torch.float64, layout=torch.sparse_bsr) # _crow_indices tensor([[[0, 3], [0, 3], [0, 3]], [[0, 3], [0, 3], [0, 3]]]) # _col_indices tensor([[[0, 1, 2], [0, 1, 2], [0, 1, 2]], [[0, 1, 2], [0, 1, 2], [0, 1, 2]]]) # _values tensor([[[[[1.], [3.]], [[2.], [0.]], [[0.], [4.]]], [[[1.], [4.]], [[2.], [0.]], [[3.], [0.]]], [[[1.], [2.]], [[0.], [3.]], [[0.], [4.]]]], [[[[0.], [2.]], [[1.], [3.]], [[0.], [4.]]], [[[1.], [3.]], [[0.], [4.]], [[2.], [0.]]], [[[1.], [0.]], [[2.], [4.]], [[3.], [0.]]]]], dtype=torch.float64) ########## torch.float64/torch.int64/size=()+(8, 6)+() ########## # sparse tensor tensor(crow_indices=tensor([0, 2, 3, 5, 7]), col_indices=tensor([0, 1, 0, 0, 1, 0, 1]), values=tensor([[[ 0., 1., 0.], [ 0., 4., 0.]], [[ 2., 0., 3.], [ 0., 5., 0.]], [[ 6., 7., 8.], [ 0., 0., 0.]], [[ 0., 9., 0.], [13., 0., 14.]], [[10., 11., 12.], [15., 16., 17.]], [[ 0., 0., 0.], [20., 21., 22.]], [[ 0., 18., 19.], [ 0., 23., 24.]]]), size=(8, 6), nnz=7, dtype=torch.float64, layout=torch.sparse_bsr) # _crow_indices tensor([0, 2, 3, 5, 7]) # _col_indices tensor([0, 1, 0, 0, 1, 0, 1]) # _values tensor([[[ 0., 1., 0.], [ 0., 4., 0.]], [[ 2., 0., 3.], [ 0., 5., 0.]], [[ 6., 7., 8.], [ 0., 0., 0.]], [[ 0., 9., 0.], [13., 0., 14.]], [[10., 11., 12.], [15., 16., 17.]], [[ 0., 0., 0.], [20., 21., 22.]], [[ 0., 18., 19.], [ 0., 23., 24.]]], dtype=torch.float64) ########## torch.float32/torch.int32/size=(2, 3)+(2, 3)+(4,) ########## # sparse tensor tensor(crow_indices=tensor([[[0, 3], [0, 3], [0, 3]], [[0, 3], [0, 3], [0, 3]]]), col_indices=tensor([[[0, 1, 2], [0, 1, 2], [0, 1, 2]], [[0, 1, 2], [0, 1, 2], [0, 1, 2]]]), values=tensor([[[[[[1., 2., 3., 4.]], [[3., 4., 5., 6.]]], [[[2., 3., 4., 5.]], [[0., 0., 0., 0.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]]], [[[[1., 2., 3., 4.]], [[4., 5., 6., 7.]]], [[[2., 3., 4., 5.]], [[0., 0., 0., 0.]]], [[[3., 4., 5., 6.]], [[0., 0., 0., 0.]]]], [[[[1., 2., 3., 4.]], [[2., 3., 4., 5.]]], [[[0., 0., 0., 0.]], [[3., 4., 5., 6.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]]]], [[[[[0., 0., 0., 0.]], [[2., 3., 4., 5.]]], [[[1., 2., 3., 4.]], [[3., 4., 5., 6.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]]], [[[[1., 2., 3., 4.]], [[3., 4., 5., 6.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]], [[[2., 3., 4., 5.]], [[0., 0., 0., 0.]]]], [[[[1., 2., 3., 4.]], [[0., 0., 0., 0.]]], [[[2., 3., 4., 5.]], [[4., 5., 6., 7.]]], [[[3., 4., 5., 6.]], [[0., 0., 0., 0.]]]]]]), size=(2, 3, 2, 3, 4), nnz=3, layout=torch.sparse_bsr) # _crow_indices tensor([[[0, 3], [0, 3], [0, 3]], [[0, 3], [0, 3], [0, 3]]], dtype=torch.int32) # _col_indices tensor([[[0, 1, 2], [0, 1, 2], [0, 1, 2]], [[0, 1, 2], [0, 1, 2], [0, 1, 2]]], dtype=torch.int32) # _values tensor([[[[[[1., 2., 3., 4.]], [[3., 4., 5., 6.]]], [[[2., 3., 4., 5.]], [[0., 0., 0., 0.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]]], [[[[1., 2., 3., 4.]], [[4., 5., 6., 7.]]], [[[2., 3., 4., 5.]], [[0., 0., 0., 0.]]], [[[3., 4., 5., 6.]], [[0., 0., 0., 0.]]]], [[[[1., 2., 3., 4.]], [[2., 3., 4., 5.]]], [[[0., 0., 0., 0.]], [[3., 4., 5., 6.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]]]], [[[[[0., 0., 0., 0.]], [[2., 3., 4., 5.]]], [[[1., 2., 3., 4.]], [[3., 4., 5., 6.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]]], [[[[1., 2., 3., 4.]], [[3., 4., 5., 6.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]], [[[2., 3., 4., 5.]], [[0., 0., 0., 0.]]]], [[[[1., 2., 3., 4.]], [[0., 0., 0., 0.]]], [[[2., 3., 4., 5.]], [[4., 5., 6., 7.]]], [[[3., 4., 5., 6.]], [[0., 0., 0., 0.]]]]]]) ########## torch.float32/torch.int32/size=()+(8, 6)+(4, 2) ########## # sparse tensor tensor(crow_indices=tensor([0, 2, 3, 5, 7]), col_indices=tensor([0, 1, 0, 0, 1, 0, 1]), values=tensor([[[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 1., 11.], [ 2., 12.], [ 3., 13.], [ 4., 14.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 4., 14.], [ 5., 15.], [ 6., 16.], [ 7., 17.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]]], [[[[ 2., 12.], [ 3., 13.], [ 4., 14.], [ 5., 15.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 3., 13.], [ 4., 14.], [ 5., 15.], [ 6., 16.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 5., 15.], [ 6., 16.], [ 7., 17.], [ 8., 18.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]]], [[[[ 6., 16.], [ 7., 17.], [ 8., 18.], [ 9., 19.]], [[ 7., 17.], [ 8., 18.], [ 9., 19.], [10., 20.]], [[ 8., 18.], [ 9., 19.], [10., 20.], [11., 21.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]]], [[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 9., 19.], [10., 20.], [11., 21.], [12., 22.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]], [[[13., 23.], [14., 24.], [15., 25.], [16., 26.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[14., 24.], [15., 25.], [16., 26.], [17., 27.]]]], [[[[10., 20.], [11., 21.], [12., 22.], [13., 23.]], [[11., 21.], [12., 22.], [13., 23.], [14., 24.]], [[12., 22.], [13., 23.], [14., 24.], [15., 25.]]], [[[15., 25.], [16., 26.], [17., 27.], [18., 28.]], [[16., 26.], [17., 27.], [18., 28.], [19., 29.]], [[17., 27.], [18., 28.], [19., 29.], [20., 30.]]]], [[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]], [[[20., 30.], [21., 31.], [22., 32.], [23., 33.]], [[21., 31.], [22., 32.], [23., 33.], [24., 34.]], [[22., 32.], [23., 33.], [24., 34.], [25., 35.]]]], [[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[18., 28.], [19., 29.], [20., 30.], [21., 31.]], [[19., 29.], [20., 30.], [21., 31.], [22., 32.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[23., 33.], [24., 34.], [25., 35.], [26., 36.]], [[24., 34.], [25., 35.], [26., 36.], [27., 37.]]]]]), size=(8, 6, 4, 2), nnz=7, layout=torch.sparse_bsr) # _crow_indices tensor([0, 2, 3, 5, 7], dtype=torch.int32) # _col_indices tensor([0, 1, 0, 0, 1, 0, 1], dtype=torch.int32) # _values tensor([[[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 1., 11.], [ 2., 12.], [ 3., 13.], [ 4., 14.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 4., 14.], [ 5., 15.], [ 6., 16.], [ 7., 17.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]]], [[[[ 2., 12.], [ 3., 13.], [ 4., 14.], [ 5., 15.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 3., 13.], [ 4., 14.], [ 5., 15.], [ 6., 16.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 5., 15.], [ 6., 16.], [ 7., 17.], [ 8., 18.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]]], [[[[ 6., 16.], [ 7., 17.], [ 8., 18.], [ 9., 19.]], [[ 7., 17.], [ 8., 18.], [ 9., 19.], [10., 20.]], [[ 8., 18.], [ 9., 19.], [10., 20.], [11., 21.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]]], [[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 9., 19.], [10., 20.], [11., 21.], [12., 22.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]], [[[13., 23.], [14., 24.], [15., 25.], [16., 26.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[14., 24.], [15., 25.], [16., 26.], [17., 27.]]]], [[[[10., 20.], [11., 21.], [12., 22.], [13., 23.]], [[11., 21.], [12., 22.], [13., 23.], [14., 24.]], [[12., 22.], [13., 23.], [14., 24.], [15., 25.]]], [[[15., 25.], [16., 26.], [17., 27.], [18., 28.]], [[16., 26.], [17., 27.], [18., 28.], [19., 29.]], [[17., 27.], [18., 28.], [19., 29.], [20., 30.]]]], [[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]], [[[20., 30.], [21., 31.], [22., 32.], [23., 33.]], [[21., 31.], [22., 32.], [23., 33.], [24., 34.]], [[22., 32.], [23., 33.], [24., 34.], [25., 35.]]]], [[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[18., 28.], [19., 29.], [20., 30.], [21., 31.]], [[19., 29.], [20., 30.], [21., 31.], [22., 32.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[23., 33.], [24., 34.], [25., 35.], [26., 36.]], [[24., 34.], [25., 35.], [26., 36.], [27., 37.]]]]]) ########## torch.float64/torch.int32/size=(2, 3)+(2, 3)+(4,) ########## # sparse tensor tensor(crow_indices=tensor([[[0, 3], [0, 3], [0, 3]], [[0, 3], [0, 3], [0, 3]]]), col_indices=tensor([[[0, 1, 2], [0, 1, 2], [0, 1, 2]], [[0, 1, 2], [0, 1, 2], [0, 1, 2]]]), values=tensor([[[[[[1., 2., 3., 4.]], [[3., 4., 5., 6.]]], [[[2., 3., 4., 5.]], [[0., 0., 0., 0.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]]], [[[[1., 2., 3., 4.]], [[4., 5., 6., 7.]]], [[[2., 3., 4., 5.]], [[0., 0., 0., 0.]]], [[[3., 4., 5., 6.]], [[0., 0., 0., 0.]]]], [[[[1., 2., 3., 4.]], [[2., 3., 4., 5.]]], [[[0., 0., 0., 0.]], [[3., 4., 5., 6.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]]]], [[[[[0., 0., 0., 0.]], [[2., 3., 4., 5.]]], [[[1., 2., 3., 4.]], [[3., 4., 5., 6.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]]], [[[[1., 2., 3., 4.]], [[3., 4., 5., 6.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]], [[[2., 3., 4., 5.]], [[0., 0., 0., 0.]]]], [[[[1., 2., 3., 4.]], [[0., 0., 0., 0.]]], [[[2., 3., 4., 5.]], [[4., 5., 6., 7.]]], [[[3., 4., 5., 6.]], [[0., 0., 0., 0.]]]]]]), size=(2, 3, 2, 3, 4), nnz=3, dtype=torch.float64, layout=torch.sparse_bsr) # _crow_indices tensor([[[0, 3], [0, 3], [0, 3]], [[0, 3], [0, 3], [0, 3]]], dtype=torch.int32) # _col_indices tensor([[[0, 1, 2], [0, 1, 2], [0, 1, 2]], [[0, 1, 2], [0, 1, 2], [0, 1, 2]]], dtype=torch.int32) # _values tensor([[[[[[1., 2., 3., 4.]], [[3., 4., 5., 6.]]], [[[2., 3., 4., 5.]], [[0., 0., 0., 0.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]]], [[[[1., 2., 3., 4.]], [[4., 5., 6., 7.]]], [[[2., 3., 4., 5.]], [[0., 0., 0., 0.]]], [[[3., 4., 5., 6.]], [[0., 0., 0., 0.]]]], [[[[1., 2., 3., 4.]], [[2., 3., 4., 5.]]], [[[0., 0., 0., 0.]], [[3., 4., 5., 6.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]]]], [[[[[0., 0., 0., 0.]], [[2., 3., 4., 5.]]], [[[1., 2., 3., 4.]], [[3., 4., 5., 6.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]]], [[[[1., 2., 3., 4.]], [[3., 4., 5., 6.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]], [[[2., 3., 4., 5.]], [[0., 0., 0., 0.]]]], [[[[1., 2., 3., 4.]], [[0., 0., 0., 0.]]], [[[2., 3., 4., 5.]], [[4., 5., 6., 7.]]], [[[3., 4., 5., 6.]], [[0., 0., 0., 0.]]]]]], dtype=torch.float64) ########## torch.float64/torch.int32/size=()+(8, 6)+(4, 2) ########## # sparse tensor tensor(crow_indices=tensor([0, 2, 3, 5, 7]), col_indices=tensor([0, 1, 0, 0, 1, 0, 1]), values=tensor([[[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 1., 11.], [ 2., 12.], [ 3., 13.], [ 4., 14.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 4., 14.], [ 5., 15.], [ 6., 16.], [ 7., 17.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]]], [[[[ 2., 12.], [ 3., 13.], [ 4., 14.], [ 5., 15.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 3., 13.], [ 4., 14.], [ 5., 15.], [ 6., 16.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 5., 15.], [ 6., 16.], [ 7., 17.], [ 8., 18.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]]], [[[[ 6., 16.], [ 7., 17.], [ 8., 18.], [ 9., 19.]], [[ 7., 17.], [ 8., 18.], [ 9., 19.], [10., 20.]], [[ 8., 18.], [ 9., 19.], [10., 20.], [11., 21.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]]], [[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 9., 19.], [10., 20.], [11., 21.], [12., 22.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]], [[[13., 23.], [14., 24.], [15., 25.], [16., 26.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[14., 24.], [15., 25.], [16., 26.], [17., 27.]]]], [[[[10., 20.], [11., 21.], [12., 22.], [13., 23.]], [[11., 21.], [12., 22.], [13., 23.], [14., 24.]], [[12., 22.], [13., 23.], [14., 24.], [15., 25.]]], [[[15., 25.], [16., 26.], [17., 27.], [18., 28.]], [[16., 26.], [17., 27.], [18., 28.], [19., 29.]], [[17., 27.], [18., 28.], [19., 29.], [20., 30.]]]], [[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]], [[[20., 30.], [21., 31.], [22., 32.], [23., 33.]], [[21., 31.], [22., 32.], [23., 33.], [24., 34.]], [[22., 32.], [23., 33.], [24., 34.], [25., 35.]]]], [[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[18., 28.], [19., 29.], [20., 30.], [21., 31.]], [[19., 29.], [20., 30.], [21., 31.], [22., 32.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[23., 33.], [24., 34.], [25., 35.], [26., 36.]], [[24., 34.], [25., 35.], [26., 36.], [27., 37.]]]]]), size=(8, 6, 4, 2), nnz=7, dtype=torch.float64, layout=torch.sparse_bsr) # _crow_indices tensor([0, 2, 3, 5, 7], dtype=torch.int32) # _col_indices tensor([0, 1, 0, 0, 1, 0, 1], dtype=torch.int32) # _values tensor([[[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 1., 11.], [ 2., 12.], [ 3., 13.], [ 4., 14.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 4., 14.], [ 5., 15.], [ 6., 16.], [ 7., 17.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]]], [[[[ 2., 12.], [ 3., 13.], [ 4., 14.], [ 5., 15.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 3., 13.], [ 4., 14.], [ 5., 15.], [ 6., 16.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 5., 15.], [ 6., 16.], [ 7., 17.], [ 8., 18.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]]], [[[[ 6., 16.], [ 7., 17.], [ 8., 18.], [ 9., 19.]], [[ 7., 17.], [ 8., 18.], [ 9., 19.], [10., 20.]], [[ 8., 18.], [ 9., 19.], [10., 20.], [11., 21.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]]], [[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 9., 19.], [10., 20.], [11., 21.], [12., 22.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]], [[[13., 23.], [14., 24.], [15., 25.], [16., 26.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[14., 24.], [15., 25.], [16., 26.], [17., 27.]]]], [[[[10., 20.], [11., 21.], [12., 22.], [13., 23.]], [[11., 21.], [12., 22.], [13., 23.], [14., 24.]], [[12., 22.], [13., 23.], [14., 24.], [15., 25.]]], [[[15., 25.], [16., 26.], [17., 27.], [18., 28.]], [[16., 26.], [17., 27.], [18., 28.], [19., 29.]], [[17., 27.], [18., 28.], [19., 29.], [20., 30.]]]], [[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]], [[[20., 30.], [21., 31.], [22., 32.], [23., 33.]], [[21., 31.], [22., 32.], [23., 33.], [24., 34.]], [[22., 32.], [23., 33.], [24., 34.], [25., 35.]]]], [[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[18., 28.], [19., 29.], [20., 30.], [21., 31.]], [[19., 29.], [20., 30.], [21., 31.], [22., 32.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[23., 33.], [24., 34.], [25., 35.], [26., 36.]], [[24., 34.], [25., 35.], [26., 36.], [27., 37.]]]]], dtype=torch.float64) ########## torch.float32/torch.int64/size=(2, 3)+(2, 3)+(4,) ########## # sparse tensor tensor(crow_indices=tensor([[[0, 3], [0, 3], [0, 3]], [[0, 3], [0, 3], [0, 3]]]), col_indices=tensor([[[0, 1, 2], [0, 1, 2], [0, 1, 2]], [[0, 1, 2], [0, 1, 2], [0, 1, 2]]]), values=tensor([[[[[[1., 2., 3., 4.]], [[3., 4., 5., 6.]]], [[[2., 3., 4., 5.]], [[0., 0., 0., 0.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]]], [[[[1., 2., 3., 4.]], [[4., 5., 6., 7.]]], [[[2., 3., 4., 5.]], [[0., 0., 0., 0.]]], [[[3., 4., 5., 6.]], [[0., 0., 0., 0.]]]], [[[[1., 2., 3., 4.]], [[2., 3., 4., 5.]]], [[[0., 0., 0., 0.]], [[3., 4., 5., 6.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]]]], [[[[[0., 0., 0., 0.]], [[2., 3., 4., 5.]]], [[[1., 2., 3., 4.]], [[3., 4., 5., 6.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]]], [[[[1., 2., 3., 4.]], [[3., 4., 5., 6.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]], [[[2., 3., 4., 5.]], [[0., 0., 0., 0.]]]], [[[[1., 2., 3., 4.]], [[0., 0., 0., 0.]]], [[[2., 3., 4., 5.]], [[4., 5., 6., 7.]]], [[[3., 4., 5., 6.]], [[0., 0., 0., 0.]]]]]]), size=(2, 3, 2, 3, 4), nnz=3, layout=torch.sparse_bsr) # _crow_indices tensor([[[0, 3], [0, 3], [0, 3]], [[0, 3], [0, 3], [0, 3]]]) # _col_indices tensor([[[0, 1, 2], [0, 1, 2], [0, 1, 2]], [[0, 1, 2], [0, 1, 2], [0, 1, 2]]]) # _values tensor([[[[[[1., 2., 3., 4.]], [[3., 4., 5., 6.]]], [[[2., 3., 4., 5.]], [[0., 0., 0., 0.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]]], [[[[1., 2., 3., 4.]], [[4., 5., 6., 7.]]], [[[2., 3., 4., 5.]], [[0., 0., 0., 0.]]], [[[3., 4., 5., 6.]], [[0., 0., 0., 0.]]]], [[[[1., 2., 3., 4.]], [[2., 3., 4., 5.]]], [[[0., 0., 0., 0.]], [[3., 4., 5., 6.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]]]], [[[[[0., 0., 0., 0.]], [[2., 3., 4., 5.]]], [[[1., 2., 3., 4.]], [[3., 4., 5., 6.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]]], [[[[1., 2., 3., 4.]], [[3., 4., 5., 6.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]], [[[2., 3., 4., 5.]], [[0., 0., 0., 0.]]]], [[[[1., 2., 3., 4.]], [[0., 0., 0., 0.]]], [[[2., 3., 4., 5.]], [[4., 5., 6., 7.]]], [[[3., 4., 5., 6.]], [[0., 0., 0., 0.]]]]]]) ########## torch.float32/torch.int64/size=()+(8, 6)+(4, 2) ########## # sparse tensor tensor(crow_indices=tensor([0, 2, 3, 5, 7]), col_indices=tensor([0, 1, 0, 0, 1, 0, 1]), values=tensor([[[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 1., 11.], [ 2., 12.], [ 3., 13.], [ 4., 14.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 4., 14.], [ 5., 15.], [ 6., 16.], [ 7., 17.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]]], [[[[ 2., 12.], [ 3., 13.], [ 4., 14.], [ 5., 15.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 3., 13.], [ 4., 14.], [ 5., 15.], [ 6., 16.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 5., 15.], [ 6., 16.], [ 7., 17.], [ 8., 18.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]]], [[[[ 6., 16.], [ 7., 17.], [ 8., 18.], [ 9., 19.]], [[ 7., 17.], [ 8., 18.], [ 9., 19.], [10., 20.]], [[ 8., 18.], [ 9., 19.], [10., 20.], [11., 21.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]]], [[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 9., 19.], [10., 20.], [11., 21.], [12., 22.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]], [[[13., 23.], [14., 24.], [15., 25.], [16., 26.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[14., 24.], [15., 25.], [16., 26.], [17., 27.]]]], [[[[10., 20.], [11., 21.], [12., 22.], [13., 23.]], [[11., 21.], [12., 22.], [13., 23.], [14., 24.]], [[12., 22.], [13., 23.], [14., 24.], [15., 25.]]], [[[15., 25.], [16., 26.], [17., 27.], [18., 28.]], [[16., 26.], [17., 27.], [18., 28.], [19., 29.]], [[17., 27.], [18., 28.], [19., 29.], [20., 30.]]]], [[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]], [[[20., 30.], [21., 31.], [22., 32.], [23., 33.]], [[21., 31.], [22., 32.], [23., 33.], [24., 34.]], [[22., 32.], [23., 33.], [24., 34.], [25., 35.]]]], [[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[18., 28.], [19., 29.], [20., 30.], [21., 31.]], [[19., 29.], [20., 30.], [21., 31.], [22., 32.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[23., 33.], [24., 34.], [25., 35.], [26., 36.]], [[24., 34.], [25., 35.], [26., 36.], [27., 37.]]]]]), size=(8, 6, 4, 2), nnz=7, layout=torch.sparse_bsr) # _crow_indices tensor([0, 2, 3, 5, 7]) # _col_indices tensor([0, 1, 0, 0, 1, 0, 1]) # _values tensor([[[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 1., 11.], [ 2., 12.], [ 3., 13.], [ 4., 14.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 4., 14.], [ 5., 15.], [ 6., 16.], [ 7., 17.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]]], [[[[ 2., 12.], [ 3., 13.], [ 4., 14.], [ 5., 15.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 3., 13.], [ 4., 14.], [ 5., 15.], [ 6., 16.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 5., 15.], [ 6., 16.], [ 7., 17.], [ 8., 18.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]]], [[[[ 6., 16.], [ 7., 17.], [ 8., 18.], [ 9., 19.]], [[ 7., 17.], [ 8., 18.], [ 9., 19.], [10., 20.]], [[ 8., 18.], [ 9., 19.], [10., 20.], [11., 21.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]]], [[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 9., 19.], [10., 20.], [11., 21.], [12., 22.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]], [[[13., 23.], [14., 24.], [15., 25.], [16., 26.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[14., 24.], [15., 25.], [16., 26.], [17., 27.]]]], [[[[10., 20.], [11., 21.], [12., 22.], [13., 23.]], [[11., 21.], [12., 22.], [13., 23.], [14., 24.]], [[12., 22.], [13., 23.], [14., 24.], [15., 25.]]], [[[15., 25.], [16., 26.], [17., 27.], [18., 28.]], [[16., 26.], [17., 27.], [18., 28.], [19., 29.]], [[17., 27.], [18., 28.], [19., 29.], [20., 30.]]]], [[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]], [[[20., 30.], [21., 31.], [22., 32.], [23., 33.]], [[21., 31.], [22., 32.], [23., 33.], [24., 34.]], [[22., 32.], [23., 33.], [24., 34.], [25., 35.]]]], [[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[18., 28.], [19., 29.], [20., 30.], [21., 31.]], [[19., 29.], [20., 30.], [21., 31.], [22., 32.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[23., 33.], [24., 34.], [25., 35.], [26., 36.]], [[24., 34.], [25., 35.], [26., 36.], [27., 37.]]]]]) ########## torch.float64/torch.int64/size=(2, 3)+(2, 3)+(4,) ########## # sparse tensor tensor(crow_indices=tensor([[[0, 3], [0, 3], [0, 3]], [[0, 3], [0, 3], [0, 3]]]), col_indices=tensor([[[0, 1, 2], [0, 1, 2], [0, 1, 2]], [[0, 1, 2], [0, 1, 2], [0, 1, 2]]]), values=tensor([[[[[[1., 2., 3., 4.]], [[3., 4., 5., 6.]]], [[[2., 3., 4., 5.]], [[0., 0., 0., 0.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]]], [[[[1., 2., 3., 4.]], [[4., 5., 6., 7.]]], [[[2., 3., 4., 5.]], [[0., 0., 0., 0.]]], [[[3., 4., 5., 6.]], [[0., 0., 0., 0.]]]], [[[[1., 2., 3., 4.]], [[2., 3., 4., 5.]]], [[[0., 0., 0., 0.]], [[3., 4., 5., 6.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]]]], [[[[[0., 0., 0., 0.]], [[2., 3., 4., 5.]]], [[[1., 2., 3., 4.]], [[3., 4., 5., 6.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]]], [[[[1., 2., 3., 4.]], [[3., 4., 5., 6.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]], [[[2., 3., 4., 5.]], [[0., 0., 0., 0.]]]], [[[[1., 2., 3., 4.]], [[0., 0., 0., 0.]]], [[[2., 3., 4., 5.]], [[4., 5., 6., 7.]]], [[[3., 4., 5., 6.]], [[0., 0., 0., 0.]]]]]]), size=(2, 3, 2, 3, 4), nnz=3, dtype=torch.float64, layout=torch.sparse_bsr) # _crow_indices tensor([[[0, 3], [0, 3], [0, 3]], [[0, 3], [0, 3], [0, 3]]]) # _col_indices tensor([[[0, 1, 2], [0, 1, 2], [0, 1, 2]], [[0, 1, 2], [0, 1, 2], [0, 1, 2]]]) # _values tensor([[[[[[1., 2., 3., 4.]], [[3., 4., 5., 6.]]], [[[2., 3., 4., 5.]], [[0., 0., 0., 0.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]]], [[[[1., 2., 3., 4.]], [[4., 5., 6., 7.]]], [[[2., 3., 4., 5.]], [[0., 0., 0., 0.]]], [[[3., 4., 5., 6.]], [[0., 0., 0., 0.]]]], [[[[1., 2., 3., 4.]], [[2., 3., 4., 5.]]], [[[0., 0., 0., 0.]], [[3., 4., 5., 6.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]]]], [[[[[0., 0., 0., 0.]], [[2., 3., 4., 5.]]], [[[1., 2., 3., 4.]], [[3., 4., 5., 6.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]]], [[[[1., 2., 3., 4.]], [[3., 4., 5., 6.]]], [[[0., 0., 0., 0.]], [[4., 5., 6., 7.]]], [[[2., 3., 4., 5.]], [[0., 0., 0., 0.]]]], [[[[1., 2., 3., 4.]], [[0., 0., 0., 0.]]], [[[2., 3., 4., 5.]], [[4., 5., 6., 7.]]], [[[3., 4., 5., 6.]], [[0., 0., 0., 0.]]]]]], dtype=torch.float64) ########## torch.float64/torch.int64/size=()+(8, 6)+(4, 2) ########## # sparse tensor tensor(crow_indices=tensor([0, 2, 3, 5, 7]), col_indices=tensor([0, 1, 0, 0, 1, 0, 1]), values=tensor([[[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 1., 11.], [ 2., 12.], [ 3., 13.], [ 4., 14.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 4., 14.], [ 5., 15.], [ 6., 16.], [ 7., 17.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]]], [[[[ 2., 12.], [ 3., 13.], [ 4., 14.], [ 5., 15.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 3., 13.], [ 4., 14.], [ 5., 15.], [ 6., 16.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 5., 15.], [ 6., 16.], [ 7., 17.], [ 8., 18.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]]], [[[[ 6., 16.], [ 7., 17.], [ 8., 18.], [ 9., 19.]], [[ 7., 17.], [ 8., 18.], [ 9., 19.], [10., 20.]], [[ 8., 18.], [ 9., 19.], [10., 20.], [11., 21.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]]], [[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 9., 19.], [10., 20.], [11., 21.], [12., 22.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]], [[[13., 23.], [14., 24.], [15., 25.], [16., 26.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[14., 24.], [15., 25.], [16., 26.], [17., 27.]]]], [[[[10., 20.], [11., 21.], [12., 22.], [13., 23.]], [[11., 21.], [12., 22.], [13., 23.], [14., 24.]], [[12., 22.], [13., 23.], [14., 24.], [15., 25.]]], [[[15., 25.], [16., 26.], [17., 27.], [18., 28.]], [[16., 26.], [17., 27.], [18., 28.], [19., 29.]], [[17., 27.], [18., 28.], [19., 29.], [20., 30.]]]], [[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]], [[[20., 30.], [21., 31.], [22., 32.], [23., 33.]], [[21., 31.], [22., 32.], [23., 33.], [24., 34.]], [[22., 32.], [23., 33.], [24., 34.], [25., 35.]]]], [[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[18., 28.], [19., 29.], [20., 30.], [21., 31.]], [[19., 29.], [20., 30.], [21., 31.], [22., 32.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[23., 33.], [24., 34.], [25., 35.], [26., 36.]], [[24., 34.], [25., 35.], [26., 36.], [27., 37.]]]]]), size=(8, 6, 4, 2), nnz=7, dtype=torch.float64, layout=torch.sparse_bsr) # _crow_indices tensor([0, 2, 3, 5, 7]) # _col_indices tensor([0, 1, 0, 0, 1, 0, 1]) # _values tensor([[[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 1., 11.], [ 2., 12.], [ 3., 13.], [ 4., 14.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 4., 14.], [ 5., 15.], [ 6., 16.], [ 7., 17.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]]], [[[[ 2., 12.], [ 3., 13.], [ 4., 14.], [ 5., 15.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 3., 13.], [ 4., 14.], [ 5., 15.], [ 6., 16.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 5., 15.], [ 6., 16.], [ 7., 17.], [ 8., 18.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]]], [[[[ 6., 16.], [ 7., 17.], [ 8., 18.], [ 9., 19.]], [[ 7., 17.], [ 8., 18.], [ 9., 19.], [10., 20.]], [[ 8., 18.], [ 9., 19.], [10., 20.], [11., 21.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]]], [[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 9., 19.], [10., 20.], [11., 21.], [12., 22.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]], [[[13., 23.], [14., 24.], [15., 25.], [16., 26.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[14., 24.], [15., 25.], [16., 26.], [17., 27.]]]], [[[[10., 20.], [11., 21.], [12., 22.], [13., 23.]], [[11., 21.], [12., 22.], [13., 23.], [14., 24.]], [[12., 22.], [13., 23.], [14., 24.], [15., 25.]]], [[[15., 25.], [16., 26.], [17., 27.], [18., 28.]], [[16., 26.], [17., 27.], [18., 28.], [19., 29.]], [[17., 27.], [18., 28.], [19., 29.], [20., 30.]]]], [[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]]], [[[20., 30.], [21., 31.], [22., 32.], [23., 33.]], [[21., 31.], [22., 32.], [23., 33.], [24., 34.]], [[22., 32.], [23., 33.], [24., 34.], [25., 35.]]]], [[[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[18., 28.], [19., 29.], [20., 30.], [21., 31.]], [[19., 29.], [20., 30.], [21., 31.], [22., 32.]]], [[[ 0., 0.], [ 0., 0.], [ 0., 0.], [ 0., 0.]], [[23., 33.], [24., 34.], [25., 35.], [26., 36.]], [[24., 34.], [25., 35.], [26., 36.], [27., 37.]]]]], dtype=torch.float64)