TestSparseCompressedCUDA.test_print_SparseBSR_cuda.expect

########## 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.]]]]]), device='cuda: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]]], device='cuda:0', dtype=torch.int32)
# _col_indices
tensor([[[0, 1, 2],
         [0, 1, 2],
         [0, 1, 2]],

        [[0, 1, 2],
         [0, 1, 2],
         [0, 1, 2]]], device='cuda:0', 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.]]]]], device='cuda: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.]]]), device='cuda:0', size=(8, 6), nnz=7,
       layout=torch.sparse_bsr)
# _crow_indices
tensor([0, 2, 3, 5, 7], device='cuda:0', dtype=torch.int32)
# _col_indices
tensor([0, 1, 0, 0, 1, 0, 1], device='cuda:0', 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.]]], device='cuda:0')


########## 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.]]]]]), device='cuda: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]]], device='cuda:0', dtype=torch.int32)
# _col_indices
tensor([[[0, 1, 2],
         [0, 1, 2],
         [0, 1, 2]],

        [[0, 1, 2],
         [0, 1, 2],
         [0, 1, 2]]], device='cuda:0', 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.]]]]], device='cuda: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.]]]), device='cuda:0', size=(8, 6), nnz=7,
       dtype=torch.float64, layout=torch.sparse_bsr)
# _crow_indices
tensor([0, 2, 3, 5, 7], device='cuda:0', dtype=torch.int32)
# _col_indices
tensor([0, 1, 0, 0, 1, 0, 1], device='cuda:0', 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.]]], device='cuda:0', 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.]]]]]), device='cuda: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]]], device='cuda:0')
# _col_indices
tensor([[[0, 1, 2],
         [0, 1, 2],
         [0, 1, 2]],

        [[0, 1, 2],
         [0, 1, 2],
         [0, 1, 2]]], device='cuda:0')
# _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.]]]]], device='cuda: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.]]]), device='cuda:0', size=(8, 6), nnz=7,
       layout=torch.sparse_bsr)
# _crow_indices
tensor([0, 2, 3, 5, 7], device='cuda:0')
# _col_indices
tensor([0, 1, 0, 0, 1, 0, 1], device='cuda:0')
# _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.]]], device='cuda:0')


########## 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.]]]]]), device='cuda: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]]], device='cuda:0')
# _col_indices
tensor([[[0, 1, 2],
         [0, 1, 2],
         [0, 1, 2]],

        [[0, 1, 2],
         [0, 1, 2],
         [0, 1, 2]]], device='cuda:0')
# _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.]]]]], device='cuda: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.]]]), device='cuda:0', size=(8, 6), nnz=7,
       dtype=torch.float64, layout=torch.sparse_bsr)
# _crow_indices
tensor([0, 2, 3, 5, 7], device='cuda:0')
# _col_indices
tensor([0, 1, 0, 0, 1, 0, 1], device='cuda:0')
# _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.]]], device='cuda:0', 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.]]]]]]), device='cuda: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]]], device='cuda:0', dtype=torch.int32)
# _col_indices
tensor([[[0, 1, 2],
         [0, 1, 2],
         [0, 1, 2]],

        [[0, 1, 2],
         [0, 1, 2],
         [0, 1, 2]]], device='cuda:0', 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.]]]]]], device='cuda: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.]]]]]), device='cuda:0', size=(8, 6, 4, 2),
       nnz=7, layout=torch.sparse_bsr)
# _crow_indices
tensor([0, 2, 3, 5, 7], device='cuda:0', dtype=torch.int32)
# _col_indices
tensor([0, 1, 0, 0, 1, 0, 1], device='cuda:0', 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.]]]]], device='cuda:0')


########## 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.]]]]]]), device='cuda: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]]], device='cuda:0', dtype=torch.int32)
# _col_indices
tensor([[[0, 1, 2],
         [0, 1, 2],
         [0, 1, 2]],

        [[0, 1, 2],
         [0, 1, 2],
         [0, 1, 2]]], device='cuda:0', 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.]]]]]], device='cuda: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.]]]]]), device='cuda:0', size=(8, 6, 4, 2),
       nnz=7, dtype=torch.float64, layout=torch.sparse_bsr)
# _crow_indices
tensor([0, 2, 3, 5, 7], device='cuda:0', dtype=torch.int32)
# _col_indices
tensor([0, 1, 0, 0, 1, 0, 1], device='cuda:0', 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.]]]]], device='cuda:0', 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.]]]]]]), device='cuda: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]]], device='cuda:0')
# _col_indices
tensor([[[0, 1, 2],
         [0, 1, 2],
         [0, 1, 2]],

        [[0, 1, 2],
         [0, 1, 2],
         [0, 1, 2]]], device='cuda:0')
# _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.]]]]]], device='cuda: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.]]]]]), device='cuda:0', size=(8, 6, 4, 2),
       nnz=7, layout=torch.sparse_bsr)
# _crow_indices
tensor([0, 2, 3, 5, 7], device='cuda:0')
# _col_indices
tensor([0, 1, 0, 0, 1, 0, 1], device='cuda:0')
# _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.]]]]], device='cuda:0')


########## 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.]]]]]]), device='cuda: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]]], device='cuda:0')
# _col_indices
tensor([[[0, 1, 2],
         [0, 1, 2],
         [0, 1, 2]],

        [[0, 1, 2],
         [0, 1, 2],
         [0, 1, 2]]], device='cuda:0')
# _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.]]]]]], device='cuda: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.]]]]]), device='cuda:0', size=(8, 6, 4, 2),
       nnz=7, dtype=torch.float64, layout=torch.sparse_bsr)
# _crow_indices
tensor([0, 2, 3, 5, 7], device='cuda:0')
# _col_indices
tensor([0, 1, 0, 0, 1, 0, 1], device='cuda:0')
# _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.]]]]], device='cuda:0', dtype=torch.float64)