pytorch

Форк
0
/
TestSparseCompressedCUDA.test_print_SparseBSC_cuda.expect 
3582 строки · 74.2 Кб
1
########## torch.float32/torch.int32/size=(2, 3)+(2, 3)+() ##########
2
# sparse tensor
3
tensor(ccol_indices=tensor([[[0, 1, 2, 3],
4
                             [0, 1, 2, 3],
5
                             [0, 1, 2, 3]],
6

7
                            [[0, 1, 2, 3],
8
                             [0, 1, 2, 3],
9
                             [0, 1, 2, 3]]]),
10
       row_indices=tensor([[[0, 0, 0],
11
                            [0, 0, 0],
12
                            [0, 0, 0]],
13

14
                           [[0, 0, 0],
15
                            [0, 0, 0],
16
                            [0, 0, 0]]]),
17
       values=tensor([[[[[1.],
18
                         [3.]],
19

20
                        [[2.],
21
                         [0.]],
22

23
                        [[0.],
24
                         [4.]]],
25

26

27
                       [[[1.],
28
                         [4.]],
29

30
                        [[2.],
31
                         [0.]],
32

33
                        [[3.],
34
                         [0.]]],
35

36

37
                       [[[1.],
38
                         [2.]],
39

40
                        [[0.],
41
                         [3.]],
42

43
                        [[0.],
44
                         [4.]]]],
45

46

47

48
                      [[[[0.],
49
                         [2.]],
50

51
                        [[1.],
52
                         [3.]],
53

54
                        [[0.],
55
                         [4.]]],
56

57

58
                       [[[1.],
59
                         [3.]],
60

61
                        [[0.],
62
                         [4.]],
63

64
                        [[2.],
65
                         [0.]]],
66

67

68
                       [[[1.],
69
                         [0.]],
70

71
                        [[2.],
72
                         [4.]],
73

74
                        [[3.],
75
                         [0.]]]]]), device='cuda:0', size=(2, 3, 2, 3), nnz=3,
76
       layout=torch.sparse_bsc)
77
# _ccol_indices
78
tensor([[[0, 1, 2, 3],
79
         [0, 1, 2, 3],
80
         [0, 1, 2, 3]],
81

82
        [[0, 1, 2, 3],
83
         [0, 1, 2, 3],
84
         [0, 1, 2, 3]]], device='cuda:0', dtype=torch.int32)
85
# _row_indices
86
tensor([[[0, 0, 0],
87
         [0, 0, 0],
88
         [0, 0, 0]],
89

90
        [[0, 0, 0],
91
         [0, 0, 0],
92
         [0, 0, 0]]], device='cuda:0', dtype=torch.int32)
93
# _values
94
tensor([[[[[1.],
95
           [3.]],
96

97
          [[2.],
98
           [0.]],
99

100
          [[0.],
101
           [4.]]],
102

103

104
         [[[1.],
105
           [4.]],
106

107
          [[2.],
108
           [0.]],
109

110
          [[3.],
111
           [0.]]],
112

113

114
         [[[1.],
115
           [2.]],
116

117
          [[0.],
118
           [3.]],
119

120
          [[0.],
121
           [4.]]]],
122

123

124

125
        [[[[0.],
126
           [2.]],
127

128
          [[1.],
129
           [3.]],
130

131
          [[0.],
132
           [4.]]],
133

134

135
         [[[1.],
136
           [3.]],
137

138
          [[0.],
139
           [4.]],
140

141
          [[2.],
142
           [0.]]],
143

144

145
         [[[1.],
146
           [0.]],
147

148
          [[2.],
149
           [4.]],
150

151
          [[3.],
152
           [0.]]]]], device='cuda:0')
153

154
########## torch.float32/torch.int32/size=()+(8, 6)+() ##########
155
# sparse tensor
156
tensor(ccol_indices=tensor([0, 4, 7]),
157
       row_indices=tensor([0, 1, 2, 3, 0, 2, 3]),
158
       values=tensor([[[ 0.,  1.,  0.],
159
                       [ 0.,  4.,  0.]],
160

161
                      [[ 6.,  7.,  8.],
162
                       [ 0.,  0.,  0.]],
163

164
                      [[ 0.,  9.,  0.],
165
                       [13.,  0., 14.]],
166

167
                      [[ 0.,  0.,  0.],
168
                       [20., 21., 22.]],
169

170
                      [[ 2.,  0.,  3.],
171
                       [ 0.,  5.,  0.]],
172

173
                      [[10., 11., 12.],
174
                       [15., 16., 17.]],
175

176
                      [[ 0., 18., 19.],
177
                       [ 0., 23., 24.]]]), device='cuda:0', size=(8, 6), nnz=7,
178
       layout=torch.sparse_bsc)
179
# _ccol_indices
180
tensor([0, 4, 7], device='cuda:0', dtype=torch.int32)
181
# _row_indices
182
tensor([0, 1, 2, 3, 0, 2, 3], device='cuda:0', dtype=torch.int32)
183
# _values
184
tensor([[[ 0.,  1.,  0.],
185
         [ 0.,  4.,  0.]],
186

187
        [[ 6.,  7.,  8.],
188
         [ 0.,  0.,  0.]],
189

190
        [[ 0.,  9.,  0.],
191
         [13.,  0., 14.]],
192

193
        [[ 0.,  0.,  0.],
194
         [20., 21., 22.]],
195

196
        [[ 2.,  0.,  3.],
197
         [ 0.,  5.,  0.]],
198

199
        [[10., 11., 12.],
200
         [15., 16., 17.]],
201

202
        [[ 0., 18., 19.],
203
         [ 0., 23., 24.]]], device='cuda:0')
204

205

206
########## torch.float64/torch.int32/size=(2, 3)+(2, 3)+() ##########
207
# sparse tensor
208
tensor(ccol_indices=tensor([[[0, 1, 2, 3],
209
                             [0, 1, 2, 3],
210
                             [0, 1, 2, 3]],
211

212
                            [[0, 1, 2, 3],
213
                             [0, 1, 2, 3],
214
                             [0, 1, 2, 3]]]),
215
       row_indices=tensor([[[0, 0, 0],
216
                            [0, 0, 0],
217
                            [0, 0, 0]],
218

219
                           [[0, 0, 0],
220
                            [0, 0, 0],
221
                            [0, 0, 0]]]),
222
       values=tensor([[[[[1.],
223
                         [3.]],
224

225
                        [[2.],
226
                         [0.]],
227

228
                        [[0.],
229
                         [4.]]],
230

231

232
                       [[[1.],
233
                         [4.]],
234

235
                        [[2.],
236
                         [0.]],
237

238
                        [[3.],
239
                         [0.]]],
240

241

242
                       [[[1.],
243
                         [2.]],
244

245
                        [[0.],
246
                         [3.]],
247

248
                        [[0.],
249
                         [4.]]]],
250

251

252

253
                      [[[[0.],
254
                         [2.]],
255

256
                        [[1.],
257
                         [3.]],
258

259
                        [[0.],
260
                         [4.]]],
261

262

263
                       [[[1.],
264
                         [3.]],
265

266
                        [[0.],
267
                         [4.]],
268

269
                        [[2.],
270
                         [0.]]],
271

272

273
                       [[[1.],
274
                         [0.]],
275

276
                        [[2.],
277
                         [4.]],
278

279
                        [[3.],
280
                         [0.]]]]]), device='cuda:0', size=(2, 3, 2, 3), nnz=3,
281
       dtype=torch.float64, layout=torch.sparse_bsc)
282
# _ccol_indices
283
tensor([[[0, 1, 2, 3],
284
         [0, 1, 2, 3],
285
         [0, 1, 2, 3]],
286

287
        [[0, 1, 2, 3],
288
         [0, 1, 2, 3],
289
         [0, 1, 2, 3]]], device='cuda:0', dtype=torch.int32)
290
# _row_indices
291
tensor([[[0, 0, 0],
292
         [0, 0, 0],
293
         [0, 0, 0]],
294

295
        [[0, 0, 0],
296
         [0, 0, 0],
297
         [0, 0, 0]]], device='cuda:0', dtype=torch.int32)
298
# _values
299
tensor([[[[[1.],
300
           [3.]],
301

302
          [[2.],
303
           [0.]],
304

305
          [[0.],
306
           [4.]]],
307

308

309
         [[[1.],
310
           [4.]],
311

312
          [[2.],
313
           [0.]],
314

315
          [[3.],
316
           [0.]]],
317

318

319
         [[[1.],
320
           [2.]],
321

322
          [[0.],
323
           [3.]],
324

325
          [[0.],
326
           [4.]]]],
327

328

329

330
        [[[[0.],
331
           [2.]],
332

333
          [[1.],
334
           [3.]],
335

336
          [[0.],
337
           [4.]]],
338

339

340
         [[[1.],
341
           [3.]],
342

343
          [[0.],
344
           [4.]],
345

346
          [[2.],
347
           [0.]]],
348

349

350
         [[[1.],
351
           [0.]],
352

353
          [[2.],
354
           [4.]],
355

356
          [[3.],
357
           [0.]]]]], device='cuda:0', dtype=torch.float64)
358

359
########## torch.float64/torch.int32/size=()+(8, 6)+() ##########
360
# sparse tensor
361
tensor(ccol_indices=tensor([0, 4, 7]),
362
       row_indices=tensor([0, 1, 2, 3, 0, 2, 3]),
363
       values=tensor([[[ 0.,  1.,  0.],
364
                       [ 0.,  4.,  0.]],
365

366
                      [[ 6.,  7.,  8.],
367
                       [ 0.,  0.,  0.]],
368

369
                      [[ 0.,  9.,  0.],
370
                       [13.,  0., 14.]],
371

372
                      [[ 0.,  0.,  0.],
373
                       [20., 21., 22.]],
374

375
                      [[ 2.,  0.,  3.],
376
                       [ 0.,  5.,  0.]],
377

378
                      [[10., 11., 12.],
379
                       [15., 16., 17.]],
380

381
                      [[ 0., 18., 19.],
382
                       [ 0., 23., 24.]]]), device='cuda:0', size=(8, 6), nnz=7,
383
       dtype=torch.float64, layout=torch.sparse_bsc)
384
# _ccol_indices
385
tensor([0, 4, 7], device='cuda:0', dtype=torch.int32)
386
# _row_indices
387
tensor([0, 1, 2, 3, 0, 2, 3], device='cuda:0', dtype=torch.int32)
388
# _values
389
tensor([[[ 0.,  1.,  0.],
390
         [ 0.,  4.,  0.]],
391

392
        [[ 6.,  7.,  8.],
393
         [ 0.,  0.,  0.]],
394

395
        [[ 0.,  9.,  0.],
396
         [13.,  0., 14.]],
397

398
        [[ 0.,  0.,  0.],
399
         [20., 21., 22.]],
400

401
        [[ 2.,  0.,  3.],
402
         [ 0.,  5.,  0.]],
403

404
        [[10., 11., 12.],
405
         [15., 16., 17.]],
406

407
        [[ 0., 18., 19.],
408
         [ 0., 23., 24.]]], device='cuda:0', dtype=torch.float64)
409

410

411
########## torch.float32/torch.int64/size=(2, 3)+(2, 3)+() ##########
412
# sparse tensor
413
tensor(ccol_indices=tensor([[[0, 1, 2, 3],
414
                             [0, 1, 2, 3],
415
                             [0, 1, 2, 3]],
416

417
                            [[0, 1, 2, 3],
418
                             [0, 1, 2, 3],
419
                             [0, 1, 2, 3]]]),
420
       row_indices=tensor([[[0, 0, 0],
421
                            [0, 0, 0],
422
                            [0, 0, 0]],
423

424
                           [[0, 0, 0],
425
                            [0, 0, 0],
426
                            [0, 0, 0]]]),
427
       values=tensor([[[[[1.],
428
                         [3.]],
429

430
                        [[2.],
431
                         [0.]],
432

433
                        [[0.],
434
                         [4.]]],
435

436

437
                       [[[1.],
438
                         [4.]],
439

440
                        [[2.],
441
                         [0.]],
442

443
                        [[3.],
444
                         [0.]]],
445

446

447
                       [[[1.],
448
                         [2.]],
449

450
                        [[0.],
451
                         [3.]],
452

453
                        [[0.],
454
                         [4.]]]],
455

456

457

458
                      [[[[0.],
459
                         [2.]],
460

461
                        [[1.],
462
                         [3.]],
463

464
                        [[0.],
465
                         [4.]]],
466

467

468
                       [[[1.],
469
                         [3.]],
470

471
                        [[0.],
472
                         [4.]],
473

474
                        [[2.],
475
                         [0.]]],
476

477

478
                       [[[1.],
479
                         [0.]],
480

481
                        [[2.],
482
                         [4.]],
483

484
                        [[3.],
485
                         [0.]]]]]), device='cuda:0', size=(2, 3, 2, 3), nnz=3,
486
       layout=torch.sparse_bsc)
487
# _ccol_indices
488
tensor([[[0, 1, 2, 3],
489
         [0, 1, 2, 3],
490
         [0, 1, 2, 3]],
491

492
        [[0, 1, 2, 3],
493
         [0, 1, 2, 3],
494
         [0, 1, 2, 3]]], device='cuda:0')
495
# _row_indices
496
tensor([[[0, 0, 0],
497
         [0, 0, 0],
498
         [0, 0, 0]],
499

500
        [[0, 0, 0],
501
         [0, 0, 0],
502
         [0, 0, 0]]], device='cuda:0')
503
# _values
504
tensor([[[[[1.],
505
           [3.]],
506

507
          [[2.],
508
           [0.]],
509

510
          [[0.],
511
           [4.]]],
512

513

514
         [[[1.],
515
           [4.]],
516

517
          [[2.],
518
           [0.]],
519

520
          [[3.],
521
           [0.]]],
522

523

524
         [[[1.],
525
           [2.]],
526

527
          [[0.],
528
           [3.]],
529

530
          [[0.],
531
           [4.]]]],
532

533

534

535
        [[[[0.],
536
           [2.]],
537

538
          [[1.],
539
           [3.]],
540

541
          [[0.],
542
           [4.]]],
543

544

545
         [[[1.],
546
           [3.]],
547

548
          [[0.],
549
           [4.]],
550

551
          [[2.],
552
           [0.]]],
553

554

555
         [[[1.],
556
           [0.]],
557

558
          [[2.],
559
           [4.]],
560

561
          [[3.],
562
           [0.]]]]], device='cuda:0')
563

564
########## torch.float32/torch.int64/size=()+(8, 6)+() ##########
565
# sparse tensor
566
tensor(ccol_indices=tensor([0, 4, 7]),
567
       row_indices=tensor([0, 1, 2, 3, 0, 2, 3]),
568
       values=tensor([[[ 0.,  1.,  0.],
569
                       [ 0.,  4.,  0.]],
570

571
                      [[ 6.,  7.,  8.],
572
                       [ 0.,  0.,  0.]],
573

574
                      [[ 0.,  9.,  0.],
575
                       [13.,  0., 14.]],
576

577
                      [[ 0.,  0.,  0.],
578
                       [20., 21., 22.]],
579

580
                      [[ 2.,  0.,  3.],
581
                       [ 0.,  5.,  0.]],
582

583
                      [[10., 11., 12.],
584
                       [15., 16., 17.]],
585

586
                      [[ 0., 18., 19.],
587
                       [ 0., 23., 24.]]]), device='cuda:0', size=(8, 6), nnz=7,
588
       layout=torch.sparse_bsc)
589
# _ccol_indices
590
tensor([0, 4, 7], device='cuda:0')
591
# _row_indices
592
tensor([0, 1, 2, 3, 0, 2, 3], device='cuda:0')
593
# _values
594
tensor([[[ 0.,  1.,  0.],
595
         [ 0.,  4.,  0.]],
596

597
        [[ 6.,  7.,  8.],
598
         [ 0.,  0.,  0.]],
599

600
        [[ 0.,  9.,  0.],
601
         [13.,  0., 14.]],
602

603
        [[ 0.,  0.,  0.],
604
         [20., 21., 22.]],
605

606
        [[ 2.,  0.,  3.],
607
         [ 0.,  5.,  0.]],
608

609
        [[10., 11., 12.],
610
         [15., 16., 17.]],
611

612
        [[ 0., 18., 19.],
613
         [ 0., 23., 24.]]], device='cuda:0')
614

615

616
########## torch.float64/torch.int64/size=(2, 3)+(2, 3)+() ##########
617
# sparse tensor
618
tensor(ccol_indices=tensor([[[0, 1, 2, 3],
619
                             [0, 1, 2, 3],
620
                             [0, 1, 2, 3]],
621

622
                            [[0, 1, 2, 3],
623
                             [0, 1, 2, 3],
624
                             [0, 1, 2, 3]]]),
625
       row_indices=tensor([[[0, 0, 0],
626
                            [0, 0, 0],
627
                            [0, 0, 0]],
628

629
                           [[0, 0, 0],
630
                            [0, 0, 0],
631
                            [0, 0, 0]]]),
632
       values=tensor([[[[[1.],
633
                         [3.]],
634

635
                        [[2.],
636
                         [0.]],
637

638
                        [[0.],
639
                         [4.]]],
640

641

642
                       [[[1.],
643
                         [4.]],
644

645
                        [[2.],
646
                         [0.]],
647

648
                        [[3.],
649
                         [0.]]],
650

651

652
                       [[[1.],
653
                         [2.]],
654

655
                        [[0.],
656
                         [3.]],
657

658
                        [[0.],
659
                         [4.]]]],
660

661

662

663
                      [[[[0.],
664
                         [2.]],
665

666
                        [[1.],
667
                         [3.]],
668

669
                        [[0.],
670
                         [4.]]],
671

672

673
                       [[[1.],
674
                         [3.]],
675

676
                        [[0.],
677
                         [4.]],
678

679
                        [[2.],
680
                         [0.]]],
681

682

683
                       [[[1.],
684
                         [0.]],
685

686
                        [[2.],
687
                         [4.]],
688

689
                        [[3.],
690
                         [0.]]]]]), device='cuda:0', size=(2, 3, 2, 3), nnz=3,
691
       dtype=torch.float64, layout=torch.sparse_bsc)
692
# _ccol_indices
693
tensor([[[0, 1, 2, 3],
694
         [0, 1, 2, 3],
695
         [0, 1, 2, 3]],
696

697
        [[0, 1, 2, 3],
698
         [0, 1, 2, 3],
699
         [0, 1, 2, 3]]], device='cuda:0')
700
# _row_indices
701
tensor([[[0, 0, 0],
702
         [0, 0, 0],
703
         [0, 0, 0]],
704

705
        [[0, 0, 0],
706
         [0, 0, 0],
707
         [0, 0, 0]]], device='cuda:0')
708
# _values
709
tensor([[[[[1.],
710
           [3.]],
711

712
          [[2.],
713
           [0.]],
714

715
          [[0.],
716
           [4.]]],
717

718

719
         [[[1.],
720
           [4.]],
721

722
          [[2.],
723
           [0.]],
724

725
          [[3.],
726
           [0.]]],
727

728

729
         [[[1.],
730
           [2.]],
731

732
          [[0.],
733
           [3.]],
734

735
          [[0.],
736
           [4.]]]],
737

738

739

740
        [[[[0.],
741
           [2.]],
742

743
          [[1.],
744
           [3.]],
745

746
          [[0.],
747
           [4.]]],
748

749

750
         [[[1.],
751
           [3.]],
752

753
          [[0.],
754
           [4.]],
755

756
          [[2.],
757
           [0.]]],
758

759

760
         [[[1.],
761
           [0.]],
762

763
          [[2.],
764
           [4.]],
765

766
          [[3.],
767
           [0.]]]]], device='cuda:0', dtype=torch.float64)
768

769
########## torch.float64/torch.int64/size=()+(8, 6)+() ##########
770
# sparse tensor
771
tensor(ccol_indices=tensor([0, 4, 7]),
772
       row_indices=tensor([0, 1, 2, 3, 0, 2, 3]),
773
       values=tensor([[[ 0.,  1.,  0.],
774
                       [ 0.,  4.,  0.]],
775

776
                      [[ 6.,  7.,  8.],
777
                       [ 0.,  0.,  0.]],
778

779
                      [[ 0.,  9.,  0.],
780
                       [13.,  0., 14.]],
781

782
                      [[ 0.,  0.,  0.],
783
                       [20., 21., 22.]],
784

785
                      [[ 2.,  0.,  3.],
786
                       [ 0.,  5.,  0.]],
787

788
                      [[10., 11., 12.],
789
                       [15., 16., 17.]],
790

791
                      [[ 0., 18., 19.],
792
                       [ 0., 23., 24.]]]), device='cuda:0', size=(8, 6), nnz=7,
793
       dtype=torch.float64, layout=torch.sparse_bsc)
794
# _ccol_indices
795
tensor([0, 4, 7], device='cuda:0')
796
# _row_indices
797
tensor([0, 1, 2, 3, 0, 2, 3], device='cuda:0')
798
# _values
799
tensor([[[ 0.,  1.,  0.],
800
         [ 0.,  4.,  0.]],
801

802
        [[ 6.,  7.,  8.],
803
         [ 0.,  0.,  0.]],
804

805
        [[ 0.,  9.,  0.],
806
         [13.,  0., 14.]],
807

808
        [[ 0.,  0.,  0.],
809
         [20., 21., 22.]],
810

811
        [[ 2.,  0.,  3.],
812
         [ 0.,  5.,  0.]],
813

814
        [[10., 11., 12.],
815
         [15., 16., 17.]],
816

817
        [[ 0., 18., 19.],
818
         [ 0., 23., 24.]]], device='cuda:0', dtype=torch.float64)
819

820

821
########## torch.float32/torch.int32/size=(2, 3)+(2, 3)+(4,) ##########
822
# sparse tensor
823
tensor(ccol_indices=tensor([[[0, 1, 2, 3],
824
                             [0, 1, 2, 3],
825
                             [0, 1, 2, 3]],
826

827
                            [[0, 1, 2, 3],
828
                             [0, 1, 2, 3],
829
                             [0, 1, 2, 3]]]),
830
       row_indices=tensor([[[0, 0, 0],
831
                            [0, 0, 0],
832
                            [0, 0, 0]],
833

834
                           [[0, 0, 0],
835
                            [0, 0, 0],
836
                            [0, 0, 0]]]),
837
       values=tensor([[[[[[1., 2., 3., 4.]],
838

839
                         [[3., 4., 5., 6.]]],
840

841

842
                        [[[2., 3., 4., 5.]],
843

844
                         [[0., 0., 0., 0.]]],
845

846

847
                        [[[0., 0., 0., 0.]],
848

849
                         [[4., 5., 6., 7.]]]],
850

851

852

853
                       [[[[1., 2., 3., 4.]],
854

855
                         [[4., 5., 6., 7.]]],
856

857

858
                        [[[2., 3., 4., 5.]],
859

860
                         [[0., 0., 0., 0.]]],
861

862

863
                        [[[3., 4., 5., 6.]],
864

865
                         [[0., 0., 0., 0.]]]],
866

867

868

869
                       [[[[1., 2., 3., 4.]],
870

871
                         [[2., 3., 4., 5.]]],
872

873

874
                        [[[0., 0., 0., 0.]],
875

876
                         [[3., 4., 5., 6.]]],
877

878

879
                        [[[0., 0., 0., 0.]],
880

881
                         [[4., 5., 6., 7.]]]]],
882

883

884

885

886
                      [[[[[0., 0., 0., 0.]],
887

888
                         [[2., 3., 4., 5.]]],
889

890

891
                        [[[1., 2., 3., 4.]],
892

893
                         [[3., 4., 5., 6.]]],
894

895

896
                        [[[0., 0., 0., 0.]],
897

898
                         [[4., 5., 6., 7.]]]],
899

900

901

902
                       [[[[1., 2., 3., 4.]],
903

904
                         [[3., 4., 5., 6.]]],
905

906

907
                        [[[0., 0., 0., 0.]],
908

909
                         [[4., 5., 6., 7.]]],
910

911

912
                        [[[2., 3., 4., 5.]],
913

914
                         [[0., 0., 0., 0.]]]],
915

916

917

918
                       [[[[1., 2., 3., 4.]],
919

920
                         [[0., 0., 0., 0.]]],
921

922

923
                        [[[2., 3., 4., 5.]],
924

925
                         [[4., 5., 6., 7.]]],
926

927

928
                        [[[3., 4., 5., 6.]],
929

930
                         [[0., 0., 0., 0.]]]]]]), device='cuda:0',
931
       size=(2, 3, 2, 3, 4), nnz=3, layout=torch.sparse_bsc)
932
# _ccol_indices
933
tensor([[[0, 1, 2, 3],
934
         [0, 1, 2, 3],
935
         [0, 1, 2, 3]],
936

937
        [[0, 1, 2, 3],
938
         [0, 1, 2, 3],
939
         [0, 1, 2, 3]]], device='cuda:0', dtype=torch.int32)
940
# _row_indices
941
tensor([[[0, 0, 0],
942
         [0, 0, 0],
943
         [0, 0, 0]],
944

945
        [[0, 0, 0],
946
         [0, 0, 0],
947
         [0, 0, 0]]], device='cuda:0', dtype=torch.int32)
948
# _values
949
tensor([[[[[[1., 2., 3., 4.]],
950

951
           [[3., 4., 5., 6.]]],
952

953

954
          [[[2., 3., 4., 5.]],
955

956
           [[0., 0., 0., 0.]]],
957

958

959
          [[[0., 0., 0., 0.]],
960

961
           [[4., 5., 6., 7.]]]],
962

963

964

965
         [[[[1., 2., 3., 4.]],
966

967
           [[4., 5., 6., 7.]]],
968

969

970
          [[[2., 3., 4., 5.]],
971

972
           [[0., 0., 0., 0.]]],
973

974

975
          [[[3., 4., 5., 6.]],
976

977
           [[0., 0., 0., 0.]]]],
978

979

980

981
         [[[[1., 2., 3., 4.]],
982

983
           [[2., 3., 4., 5.]]],
984

985

986
          [[[0., 0., 0., 0.]],
987

988
           [[3., 4., 5., 6.]]],
989

990

991
          [[[0., 0., 0., 0.]],
992

993
           [[4., 5., 6., 7.]]]]],
994

995

996

997

998
        [[[[[0., 0., 0., 0.]],
999

1000
           [[2., 3., 4., 5.]]],
1001

1002

1003
          [[[1., 2., 3., 4.]],
1004

1005
           [[3., 4., 5., 6.]]],
1006

1007

1008
          [[[0., 0., 0., 0.]],
1009

1010
           [[4., 5., 6., 7.]]]],
1011

1012

1013

1014
         [[[[1., 2., 3., 4.]],
1015

1016
           [[3., 4., 5., 6.]]],
1017

1018

1019
          [[[0., 0., 0., 0.]],
1020

1021
           [[4., 5., 6., 7.]]],
1022

1023

1024
          [[[2., 3., 4., 5.]],
1025

1026
           [[0., 0., 0., 0.]]]],
1027

1028

1029

1030
         [[[[1., 2., 3., 4.]],
1031

1032
           [[0., 0., 0., 0.]]],
1033

1034

1035
          [[[2., 3., 4., 5.]],
1036

1037
           [[4., 5., 6., 7.]]],
1038

1039

1040
          [[[3., 4., 5., 6.]],
1041

1042
           [[0., 0., 0., 0.]]]]]], device='cuda:0')
1043

1044
########## torch.float32/torch.int32/size=()+(8, 6)+(4, 2) ##########
1045
# sparse tensor
1046
tensor(ccol_indices=tensor([0, 4, 7]),
1047
       row_indices=tensor([0, 1, 2, 3, 0, 2, 3]),
1048
       values=tensor([[[[[ 0.,  0.],
1049
                         [ 0.,  0.],
1050
                         [ 0.,  0.],
1051
                         [ 0.,  0.]],
1052

1053
                        [[ 1., 11.],
1054
                         [ 2., 12.],
1055
                         [ 3., 13.],
1056
                         [ 4., 14.]],
1057

1058
                        [[ 0.,  0.],
1059
                         [ 0.,  0.],
1060
                         [ 0.,  0.],
1061
                         [ 0.,  0.]]],
1062

1063

1064
                       [[[ 0.,  0.],
1065
                         [ 0.,  0.],
1066
                         [ 0.,  0.],
1067
                         [ 0.,  0.]],
1068

1069
                        [[ 4., 14.],
1070
                         [ 5., 15.],
1071
                         [ 6., 16.],
1072
                         [ 7., 17.]],
1073

1074
                        [[ 0.,  0.],
1075
                         [ 0.,  0.],
1076
                         [ 0.,  0.],
1077
                         [ 0.,  0.]]]],
1078

1079

1080

1081
                      [[[[ 6., 16.],
1082
                         [ 7., 17.],
1083
                         [ 8., 18.],
1084
                         [ 9., 19.]],
1085

1086
                        [[ 7., 17.],
1087
                         [ 8., 18.],
1088
                         [ 9., 19.],
1089
                         [10., 20.]],
1090

1091
                        [[ 8., 18.],
1092
                         [ 9., 19.],
1093
                         [10., 20.],
1094
                         [11., 21.]]],
1095

1096

1097
                       [[[ 0.,  0.],
1098
                         [ 0.,  0.],
1099
                         [ 0.,  0.],
1100
                         [ 0.,  0.]],
1101

1102
                        [[ 0.,  0.],
1103
                         [ 0.,  0.],
1104
                         [ 0.,  0.],
1105
                         [ 0.,  0.]],
1106

1107
                        [[ 0.,  0.],
1108
                         [ 0.,  0.],
1109
                         [ 0.,  0.],
1110
                         [ 0.,  0.]]]],
1111

1112

1113

1114
                      [[[[ 0.,  0.],
1115
                         [ 0.,  0.],
1116
                         [ 0.,  0.],
1117
                         [ 0.,  0.]],
1118

1119
                        [[ 9., 19.],
1120
                         [10., 20.],
1121
                         [11., 21.],
1122
                         [12., 22.]],
1123

1124
                        [[ 0.,  0.],
1125
                         [ 0.,  0.],
1126
                         [ 0.,  0.],
1127
                         [ 0.,  0.]]],
1128

1129

1130
                       [[[13., 23.],
1131
                         [14., 24.],
1132
                         [15., 25.],
1133
                         [16., 26.]],
1134

1135
                        [[ 0.,  0.],
1136
                         [ 0.,  0.],
1137
                         [ 0.,  0.],
1138
                         [ 0.,  0.]],
1139

1140
                        [[14., 24.],
1141
                         [15., 25.],
1142
                         [16., 26.],
1143
                         [17., 27.]]]],
1144

1145

1146

1147
                      [[[[ 0.,  0.],
1148
                         [ 0.,  0.],
1149
                         [ 0.,  0.],
1150
                         [ 0.,  0.]],
1151

1152
                        [[ 0.,  0.],
1153
                         [ 0.,  0.],
1154
                         [ 0.,  0.],
1155
                         [ 0.,  0.]],
1156

1157
                        [[ 0.,  0.],
1158
                         [ 0.,  0.],
1159
                         [ 0.,  0.],
1160
                         [ 0.,  0.]]],
1161

1162

1163
                       [[[20., 30.],
1164
                         [21., 31.],
1165
                         [22., 32.],
1166
                         [23., 33.]],
1167

1168
                        [[21., 31.],
1169
                         [22., 32.],
1170
                         [23., 33.],
1171
                         [24., 34.]],
1172

1173
                        [[22., 32.],
1174
                         [23., 33.],
1175
                         [24., 34.],
1176
                         [25., 35.]]]],
1177

1178

1179

1180
                      [[[[ 2., 12.],
1181
                         [ 3., 13.],
1182
                         [ 4., 14.],
1183
                         [ 5., 15.]],
1184

1185
                        [[ 0.,  0.],
1186
                         [ 0.,  0.],
1187
                         [ 0.,  0.],
1188
                         [ 0.,  0.]],
1189

1190
                        [[ 3., 13.],
1191
                         [ 4., 14.],
1192
                         [ 5., 15.],
1193
                         [ 6., 16.]]],
1194

1195

1196
                       [[[ 0.,  0.],
1197
                         [ 0.,  0.],
1198
                         [ 0.,  0.],
1199
                         [ 0.,  0.]],
1200

1201
                        [[ 5., 15.],
1202
                         [ 6., 16.],
1203
                         [ 7., 17.],
1204
                         [ 8., 18.]],
1205

1206
                        [[ 0.,  0.],
1207
                         [ 0.,  0.],
1208
                         [ 0.,  0.],
1209
                         [ 0.,  0.]]]],
1210

1211

1212

1213
                      [[[[10., 20.],
1214
                         [11., 21.],
1215
                         [12., 22.],
1216
                         [13., 23.]],
1217

1218
                        [[11., 21.],
1219
                         [12., 22.],
1220
                         [13., 23.],
1221
                         [14., 24.]],
1222

1223
                        [[12., 22.],
1224
                         [13., 23.],
1225
                         [14., 24.],
1226
                         [15., 25.]]],
1227

1228

1229
                       [[[15., 25.],
1230
                         [16., 26.],
1231
                         [17., 27.],
1232
                         [18., 28.]],
1233

1234
                        [[16., 26.],
1235
                         [17., 27.],
1236
                         [18., 28.],
1237
                         [19., 29.]],
1238

1239
                        [[17., 27.],
1240
                         [18., 28.],
1241
                         [19., 29.],
1242
                         [20., 30.]]]],
1243

1244

1245

1246
                      [[[[ 0.,  0.],
1247
                         [ 0.,  0.],
1248
                         [ 0.,  0.],
1249
                         [ 0.,  0.]],
1250

1251
                        [[18., 28.],
1252
                         [19., 29.],
1253
                         [20., 30.],
1254
                         [21., 31.]],
1255

1256
                        [[19., 29.],
1257
                         [20., 30.],
1258
                         [21., 31.],
1259
                         [22., 32.]]],
1260

1261

1262
                       [[[ 0.,  0.],
1263
                         [ 0.,  0.],
1264
                         [ 0.,  0.],
1265
                         [ 0.,  0.]],
1266

1267
                        [[23., 33.],
1268
                         [24., 34.],
1269
                         [25., 35.],
1270
                         [26., 36.]],
1271

1272
                        [[24., 34.],
1273
                         [25., 35.],
1274
                         [26., 36.],
1275
                         [27., 37.]]]]]), device='cuda:0', size=(8, 6, 4, 2),
1276
       nnz=7, layout=torch.sparse_bsc)
1277
# _ccol_indices
1278
tensor([0, 4, 7], device='cuda:0', dtype=torch.int32)
1279
# _row_indices
1280
tensor([0, 1, 2, 3, 0, 2, 3], device='cuda:0', dtype=torch.int32)
1281
# _values
1282
tensor([[[[[ 0.,  0.],
1283
           [ 0.,  0.],
1284
           [ 0.,  0.],
1285
           [ 0.,  0.]],
1286

1287
          [[ 1., 11.],
1288
           [ 2., 12.],
1289
           [ 3., 13.],
1290
           [ 4., 14.]],
1291

1292
          [[ 0.,  0.],
1293
           [ 0.,  0.],
1294
           [ 0.,  0.],
1295
           [ 0.,  0.]]],
1296

1297

1298
         [[[ 0.,  0.],
1299
           [ 0.,  0.],
1300
           [ 0.,  0.],
1301
           [ 0.,  0.]],
1302

1303
          [[ 4., 14.],
1304
           [ 5., 15.],
1305
           [ 6., 16.],
1306
           [ 7., 17.]],
1307

1308
          [[ 0.,  0.],
1309
           [ 0.,  0.],
1310
           [ 0.,  0.],
1311
           [ 0.,  0.]]]],
1312

1313

1314

1315
        [[[[ 6., 16.],
1316
           [ 7., 17.],
1317
           [ 8., 18.],
1318
           [ 9., 19.]],
1319

1320
          [[ 7., 17.],
1321
           [ 8., 18.],
1322
           [ 9., 19.],
1323
           [10., 20.]],
1324

1325
          [[ 8., 18.],
1326
           [ 9., 19.],
1327
           [10., 20.],
1328
           [11., 21.]]],
1329

1330

1331
         [[[ 0.,  0.],
1332
           [ 0.,  0.],
1333
           [ 0.,  0.],
1334
           [ 0.,  0.]],
1335

1336
          [[ 0.,  0.],
1337
           [ 0.,  0.],
1338
           [ 0.,  0.],
1339
           [ 0.,  0.]],
1340

1341
          [[ 0.,  0.],
1342
           [ 0.,  0.],
1343
           [ 0.,  0.],
1344
           [ 0.,  0.]]]],
1345

1346

1347

1348
        [[[[ 0.,  0.],
1349
           [ 0.,  0.],
1350
           [ 0.,  0.],
1351
           [ 0.,  0.]],
1352

1353
          [[ 9., 19.],
1354
           [10., 20.],
1355
           [11., 21.],
1356
           [12., 22.]],
1357

1358
          [[ 0.,  0.],
1359
           [ 0.,  0.],
1360
           [ 0.,  0.],
1361
           [ 0.,  0.]]],
1362

1363

1364
         [[[13., 23.],
1365
           [14., 24.],
1366
           [15., 25.],
1367
           [16., 26.]],
1368

1369
          [[ 0.,  0.],
1370
           [ 0.,  0.],
1371
           [ 0.,  0.],
1372
           [ 0.,  0.]],
1373

1374
          [[14., 24.],
1375
           [15., 25.],
1376
           [16., 26.],
1377
           [17., 27.]]]],
1378

1379

1380

1381
        [[[[ 0.,  0.],
1382
           [ 0.,  0.],
1383
           [ 0.,  0.],
1384
           [ 0.,  0.]],
1385

1386
          [[ 0.,  0.],
1387
           [ 0.,  0.],
1388
           [ 0.,  0.],
1389
           [ 0.,  0.]],
1390

1391
          [[ 0.,  0.],
1392
           [ 0.,  0.],
1393
           [ 0.,  0.],
1394
           [ 0.,  0.]]],
1395

1396

1397
         [[[20., 30.],
1398
           [21., 31.],
1399
           [22., 32.],
1400
           [23., 33.]],
1401

1402
          [[21., 31.],
1403
           [22., 32.],
1404
           [23., 33.],
1405
           [24., 34.]],
1406

1407
          [[22., 32.],
1408
           [23., 33.],
1409
           [24., 34.],
1410
           [25., 35.]]]],
1411

1412

1413

1414
        [[[[ 2., 12.],
1415
           [ 3., 13.],
1416
           [ 4., 14.],
1417
           [ 5., 15.]],
1418

1419
          [[ 0.,  0.],
1420
           [ 0.,  0.],
1421
           [ 0.,  0.],
1422
           [ 0.,  0.]],
1423

1424
          [[ 3., 13.],
1425
           [ 4., 14.],
1426
           [ 5., 15.],
1427
           [ 6., 16.]]],
1428

1429

1430
         [[[ 0.,  0.],
1431
           [ 0.,  0.],
1432
           [ 0.,  0.],
1433
           [ 0.,  0.]],
1434

1435
          [[ 5., 15.],
1436
           [ 6., 16.],
1437
           [ 7., 17.],
1438
           [ 8., 18.]],
1439

1440
          [[ 0.,  0.],
1441
           [ 0.,  0.],
1442
           [ 0.,  0.],
1443
           [ 0.,  0.]]]],
1444

1445

1446

1447
        [[[[10., 20.],
1448
           [11., 21.],
1449
           [12., 22.],
1450
           [13., 23.]],
1451

1452
          [[11., 21.],
1453
           [12., 22.],
1454
           [13., 23.],
1455
           [14., 24.]],
1456

1457
          [[12., 22.],
1458
           [13., 23.],
1459
           [14., 24.],
1460
           [15., 25.]]],
1461

1462

1463
         [[[15., 25.],
1464
           [16., 26.],
1465
           [17., 27.],
1466
           [18., 28.]],
1467

1468
          [[16., 26.],
1469
           [17., 27.],
1470
           [18., 28.],
1471
           [19., 29.]],
1472

1473
          [[17., 27.],
1474
           [18., 28.],
1475
           [19., 29.],
1476
           [20., 30.]]]],
1477

1478

1479

1480
        [[[[ 0.,  0.],
1481
           [ 0.,  0.],
1482
           [ 0.,  0.],
1483
           [ 0.,  0.]],
1484

1485
          [[18., 28.],
1486
           [19., 29.],
1487
           [20., 30.],
1488
           [21., 31.]],
1489

1490
          [[19., 29.],
1491
           [20., 30.],
1492
           [21., 31.],
1493
           [22., 32.]]],
1494

1495

1496
         [[[ 0.,  0.],
1497
           [ 0.,  0.],
1498
           [ 0.,  0.],
1499
           [ 0.,  0.]],
1500

1501
          [[23., 33.],
1502
           [24., 34.],
1503
           [25., 35.],
1504
           [26., 36.]],
1505

1506
          [[24., 34.],
1507
           [25., 35.],
1508
           [26., 36.],
1509
           [27., 37.]]]]], device='cuda:0')
1510

1511

1512
########## torch.float64/torch.int32/size=(2, 3)+(2, 3)+(4,) ##########
1513
# sparse tensor
1514
tensor(ccol_indices=tensor([[[0, 1, 2, 3],
1515
                             [0, 1, 2, 3],
1516
                             [0, 1, 2, 3]],
1517

1518
                            [[0, 1, 2, 3],
1519
                             [0, 1, 2, 3],
1520
                             [0, 1, 2, 3]]]),
1521
       row_indices=tensor([[[0, 0, 0],
1522
                            [0, 0, 0],
1523
                            [0, 0, 0]],
1524

1525
                           [[0, 0, 0],
1526
                            [0, 0, 0],
1527
                            [0, 0, 0]]]),
1528
       values=tensor([[[[[[1., 2., 3., 4.]],
1529

1530
                         [[3., 4., 5., 6.]]],
1531

1532

1533
                        [[[2., 3., 4., 5.]],
1534

1535
                         [[0., 0., 0., 0.]]],
1536

1537

1538
                        [[[0., 0., 0., 0.]],
1539

1540
                         [[4., 5., 6., 7.]]]],
1541

1542

1543

1544
                       [[[[1., 2., 3., 4.]],
1545

1546
                         [[4., 5., 6., 7.]]],
1547

1548

1549
                        [[[2., 3., 4., 5.]],
1550

1551
                         [[0., 0., 0., 0.]]],
1552

1553

1554
                        [[[3., 4., 5., 6.]],
1555

1556
                         [[0., 0., 0., 0.]]]],
1557

1558

1559

1560
                       [[[[1., 2., 3., 4.]],
1561

1562
                         [[2., 3., 4., 5.]]],
1563

1564

1565
                        [[[0., 0., 0., 0.]],
1566

1567
                         [[3., 4., 5., 6.]]],
1568

1569

1570
                        [[[0., 0., 0., 0.]],
1571

1572
                         [[4., 5., 6., 7.]]]]],
1573

1574

1575

1576

1577
                      [[[[[0., 0., 0., 0.]],
1578

1579
                         [[2., 3., 4., 5.]]],
1580

1581

1582
                        [[[1., 2., 3., 4.]],
1583

1584
                         [[3., 4., 5., 6.]]],
1585

1586

1587
                        [[[0., 0., 0., 0.]],
1588

1589
                         [[4., 5., 6., 7.]]]],
1590

1591

1592

1593
                       [[[[1., 2., 3., 4.]],
1594

1595
                         [[3., 4., 5., 6.]]],
1596

1597

1598
                        [[[0., 0., 0., 0.]],
1599

1600
                         [[4., 5., 6., 7.]]],
1601

1602

1603
                        [[[2., 3., 4., 5.]],
1604

1605
                         [[0., 0., 0., 0.]]]],
1606

1607

1608

1609
                       [[[[1., 2., 3., 4.]],
1610

1611
                         [[0., 0., 0., 0.]]],
1612

1613

1614
                        [[[2., 3., 4., 5.]],
1615

1616
                         [[4., 5., 6., 7.]]],
1617

1618

1619
                        [[[3., 4., 5., 6.]],
1620

1621
                         [[0., 0., 0., 0.]]]]]]), device='cuda:0',
1622
       size=(2, 3, 2, 3, 4), nnz=3, dtype=torch.float64, layout=torch.sparse_bsc)
1623
# _ccol_indices
1624
tensor([[[0, 1, 2, 3],
1625
         [0, 1, 2, 3],
1626
         [0, 1, 2, 3]],
1627

1628
        [[0, 1, 2, 3],
1629
         [0, 1, 2, 3],
1630
         [0, 1, 2, 3]]], device='cuda:0', dtype=torch.int32)
1631
# _row_indices
1632
tensor([[[0, 0, 0],
1633
         [0, 0, 0],
1634
         [0, 0, 0]],
1635

1636
        [[0, 0, 0],
1637
         [0, 0, 0],
1638
         [0, 0, 0]]], device='cuda:0', dtype=torch.int32)
1639
# _values
1640
tensor([[[[[[1., 2., 3., 4.]],
1641

1642
           [[3., 4., 5., 6.]]],
1643

1644

1645
          [[[2., 3., 4., 5.]],
1646

1647
           [[0., 0., 0., 0.]]],
1648

1649

1650
          [[[0., 0., 0., 0.]],
1651

1652
           [[4., 5., 6., 7.]]]],
1653

1654

1655

1656
         [[[[1., 2., 3., 4.]],
1657

1658
           [[4., 5., 6., 7.]]],
1659

1660

1661
          [[[2., 3., 4., 5.]],
1662

1663
           [[0., 0., 0., 0.]]],
1664

1665

1666
          [[[3., 4., 5., 6.]],
1667

1668
           [[0., 0., 0., 0.]]]],
1669

1670

1671

1672
         [[[[1., 2., 3., 4.]],
1673

1674
           [[2., 3., 4., 5.]]],
1675

1676

1677
          [[[0., 0., 0., 0.]],
1678

1679
           [[3., 4., 5., 6.]]],
1680

1681

1682
          [[[0., 0., 0., 0.]],
1683

1684
           [[4., 5., 6., 7.]]]]],
1685

1686

1687

1688

1689
        [[[[[0., 0., 0., 0.]],
1690

1691
           [[2., 3., 4., 5.]]],
1692

1693

1694
          [[[1., 2., 3., 4.]],
1695

1696
           [[3., 4., 5., 6.]]],
1697

1698

1699
          [[[0., 0., 0., 0.]],
1700

1701
           [[4., 5., 6., 7.]]]],
1702

1703

1704

1705
         [[[[1., 2., 3., 4.]],
1706

1707
           [[3., 4., 5., 6.]]],
1708

1709

1710
          [[[0., 0., 0., 0.]],
1711

1712
           [[4., 5., 6., 7.]]],
1713

1714

1715
          [[[2., 3., 4., 5.]],
1716

1717
           [[0., 0., 0., 0.]]]],
1718

1719

1720

1721
         [[[[1., 2., 3., 4.]],
1722

1723
           [[0., 0., 0., 0.]]],
1724

1725

1726
          [[[2., 3., 4., 5.]],
1727

1728
           [[4., 5., 6., 7.]]],
1729

1730

1731
          [[[3., 4., 5., 6.]],
1732

1733
           [[0., 0., 0., 0.]]]]]], device='cuda:0', dtype=torch.float64)
1734

1735
########## torch.float64/torch.int32/size=()+(8, 6)+(4, 2) ##########
1736
# sparse tensor
1737
tensor(ccol_indices=tensor([0, 4, 7]),
1738
       row_indices=tensor([0, 1, 2, 3, 0, 2, 3]),
1739
       values=tensor([[[[[ 0.,  0.],
1740
                         [ 0.,  0.],
1741
                         [ 0.,  0.],
1742
                         [ 0.,  0.]],
1743

1744
                        [[ 1., 11.],
1745
                         [ 2., 12.],
1746
                         [ 3., 13.],
1747
                         [ 4., 14.]],
1748

1749
                        [[ 0.,  0.],
1750
                         [ 0.,  0.],
1751
                         [ 0.,  0.],
1752
                         [ 0.,  0.]]],
1753

1754

1755
                       [[[ 0.,  0.],
1756
                         [ 0.,  0.],
1757
                         [ 0.,  0.],
1758
                         [ 0.,  0.]],
1759

1760
                        [[ 4., 14.],
1761
                         [ 5., 15.],
1762
                         [ 6., 16.],
1763
                         [ 7., 17.]],
1764

1765
                        [[ 0.,  0.],
1766
                         [ 0.,  0.],
1767
                         [ 0.,  0.],
1768
                         [ 0.,  0.]]]],
1769

1770

1771

1772
                      [[[[ 6., 16.],
1773
                         [ 7., 17.],
1774
                         [ 8., 18.],
1775
                         [ 9., 19.]],
1776

1777
                        [[ 7., 17.],
1778
                         [ 8., 18.],
1779
                         [ 9., 19.],
1780
                         [10., 20.]],
1781

1782
                        [[ 8., 18.],
1783
                         [ 9., 19.],
1784
                         [10., 20.],
1785
                         [11., 21.]]],
1786

1787

1788
                       [[[ 0.,  0.],
1789
                         [ 0.,  0.],
1790
                         [ 0.,  0.],
1791
                         [ 0.,  0.]],
1792

1793
                        [[ 0.,  0.],
1794
                         [ 0.,  0.],
1795
                         [ 0.,  0.],
1796
                         [ 0.,  0.]],
1797

1798
                        [[ 0.,  0.],
1799
                         [ 0.,  0.],
1800
                         [ 0.,  0.],
1801
                         [ 0.,  0.]]]],
1802

1803

1804

1805
                      [[[[ 0.,  0.],
1806
                         [ 0.,  0.],
1807
                         [ 0.,  0.],
1808
                         [ 0.,  0.]],
1809

1810
                        [[ 9., 19.],
1811
                         [10., 20.],
1812
                         [11., 21.],
1813
                         [12., 22.]],
1814

1815
                        [[ 0.,  0.],
1816
                         [ 0.,  0.],
1817
                         [ 0.,  0.],
1818
                         [ 0.,  0.]]],
1819

1820

1821
                       [[[13., 23.],
1822
                         [14., 24.],
1823
                         [15., 25.],
1824
                         [16., 26.]],
1825

1826
                        [[ 0.,  0.],
1827
                         [ 0.,  0.],
1828
                         [ 0.,  0.],
1829
                         [ 0.,  0.]],
1830

1831
                        [[14., 24.],
1832
                         [15., 25.],
1833
                         [16., 26.],
1834
                         [17., 27.]]]],
1835

1836

1837

1838
                      [[[[ 0.,  0.],
1839
                         [ 0.,  0.],
1840
                         [ 0.,  0.],
1841
                         [ 0.,  0.]],
1842

1843
                        [[ 0.,  0.],
1844
                         [ 0.,  0.],
1845
                         [ 0.,  0.],
1846
                         [ 0.,  0.]],
1847

1848
                        [[ 0.,  0.],
1849
                         [ 0.,  0.],
1850
                         [ 0.,  0.],
1851
                         [ 0.,  0.]]],
1852

1853

1854
                       [[[20., 30.],
1855
                         [21., 31.],
1856
                         [22., 32.],
1857
                         [23., 33.]],
1858

1859
                        [[21., 31.],
1860
                         [22., 32.],
1861
                         [23., 33.],
1862
                         [24., 34.]],
1863

1864
                        [[22., 32.],
1865
                         [23., 33.],
1866
                         [24., 34.],
1867
                         [25., 35.]]]],
1868

1869

1870

1871
                      [[[[ 2., 12.],
1872
                         [ 3., 13.],
1873
                         [ 4., 14.],
1874
                         [ 5., 15.]],
1875

1876
                        [[ 0.,  0.],
1877
                         [ 0.,  0.],
1878
                         [ 0.,  0.],
1879
                         [ 0.,  0.]],
1880

1881
                        [[ 3., 13.],
1882
                         [ 4., 14.],
1883
                         [ 5., 15.],
1884
                         [ 6., 16.]]],
1885

1886

1887
                       [[[ 0.,  0.],
1888
                         [ 0.,  0.],
1889
                         [ 0.,  0.],
1890
                         [ 0.,  0.]],
1891

1892
                        [[ 5., 15.],
1893
                         [ 6., 16.],
1894
                         [ 7., 17.],
1895
                         [ 8., 18.]],
1896

1897
                        [[ 0.,  0.],
1898
                         [ 0.,  0.],
1899
                         [ 0.,  0.],
1900
                         [ 0.,  0.]]]],
1901

1902

1903

1904
                      [[[[10., 20.],
1905
                         [11., 21.],
1906
                         [12., 22.],
1907
                         [13., 23.]],
1908

1909
                        [[11., 21.],
1910
                         [12., 22.],
1911
                         [13., 23.],
1912
                         [14., 24.]],
1913

1914
                        [[12., 22.],
1915
                         [13., 23.],
1916
                         [14., 24.],
1917
                         [15., 25.]]],
1918

1919

1920
                       [[[15., 25.],
1921
                         [16., 26.],
1922
                         [17., 27.],
1923
                         [18., 28.]],
1924

1925
                        [[16., 26.],
1926
                         [17., 27.],
1927
                         [18., 28.],
1928
                         [19., 29.]],
1929

1930
                        [[17., 27.],
1931
                         [18., 28.],
1932
                         [19., 29.],
1933
                         [20., 30.]]]],
1934

1935

1936

1937
                      [[[[ 0.,  0.],
1938
                         [ 0.,  0.],
1939
                         [ 0.,  0.],
1940
                         [ 0.,  0.]],
1941

1942
                        [[18., 28.],
1943
                         [19., 29.],
1944
                         [20., 30.],
1945
                         [21., 31.]],
1946

1947
                        [[19., 29.],
1948
                         [20., 30.],
1949
                         [21., 31.],
1950
                         [22., 32.]]],
1951

1952

1953
                       [[[ 0.,  0.],
1954
                         [ 0.,  0.],
1955
                         [ 0.,  0.],
1956
                         [ 0.,  0.]],
1957

1958
                        [[23., 33.],
1959
                         [24., 34.],
1960
                         [25., 35.],
1961
                         [26., 36.]],
1962

1963
                        [[24., 34.],
1964
                         [25., 35.],
1965
                         [26., 36.],
1966
                         [27., 37.]]]]]), device='cuda:0', size=(8, 6, 4, 2),
1967
       nnz=7, dtype=torch.float64, layout=torch.sparse_bsc)
1968
# _ccol_indices
1969
tensor([0, 4, 7], device='cuda:0', dtype=torch.int32)
1970
# _row_indices
1971
tensor([0, 1, 2, 3, 0, 2, 3], device='cuda:0', dtype=torch.int32)
1972
# _values
1973
tensor([[[[[ 0.,  0.],
1974
           [ 0.,  0.],
1975
           [ 0.,  0.],
1976
           [ 0.,  0.]],
1977

1978
          [[ 1., 11.],
1979
           [ 2., 12.],
1980
           [ 3., 13.],
1981
           [ 4., 14.]],
1982

1983
          [[ 0.,  0.],
1984
           [ 0.,  0.],
1985
           [ 0.,  0.],
1986
           [ 0.,  0.]]],
1987

1988

1989
         [[[ 0.,  0.],
1990
           [ 0.,  0.],
1991
           [ 0.,  0.],
1992
           [ 0.,  0.]],
1993

1994
          [[ 4., 14.],
1995
           [ 5., 15.],
1996
           [ 6., 16.],
1997
           [ 7., 17.]],
1998

1999
          [[ 0.,  0.],
2000
           [ 0.,  0.],
2001
           [ 0.,  0.],
2002
           [ 0.,  0.]]]],
2003

2004

2005

2006
        [[[[ 6., 16.],
2007
           [ 7., 17.],
2008
           [ 8., 18.],
2009
           [ 9., 19.]],
2010

2011
          [[ 7., 17.],
2012
           [ 8., 18.],
2013
           [ 9., 19.],
2014
           [10., 20.]],
2015

2016
          [[ 8., 18.],
2017
           [ 9., 19.],
2018
           [10., 20.],
2019
           [11., 21.]]],
2020

2021

2022
         [[[ 0.,  0.],
2023
           [ 0.,  0.],
2024
           [ 0.,  0.],
2025
           [ 0.,  0.]],
2026

2027
          [[ 0.,  0.],
2028
           [ 0.,  0.],
2029
           [ 0.,  0.],
2030
           [ 0.,  0.]],
2031

2032
          [[ 0.,  0.],
2033
           [ 0.,  0.],
2034
           [ 0.,  0.],
2035
           [ 0.,  0.]]]],
2036

2037

2038

2039
        [[[[ 0.,  0.],
2040
           [ 0.,  0.],
2041
           [ 0.,  0.],
2042
           [ 0.,  0.]],
2043

2044
          [[ 9., 19.],
2045
           [10., 20.],
2046
           [11., 21.],
2047
           [12., 22.]],
2048

2049
          [[ 0.,  0.],
2050
           [ 0.,  0.],
2051
           [ 0.,  0.],
2052
           [ 0.,  0.]]],
2053

2054

2055
         [[[13., 23.],
2056
           [14., 24.],
2057
           [15., 25.],
2058
           [16., 26.]],
2059

2060
          [[ 0.,  0.],
2061
           [ 0.,  0.],
2062
           [ 0.,  0.],
2063
           [ 0.,  0.]],
2064

2065
          [[14., 24.],
2066
           [15., 25.],
2067
           [16., 26.],
2068
           [17., 27.]]]],
2069

2070

2071

2072
        [[[[ 0.,  0.],
2073
           [ 0.,  0.],
2074
           [ 0.,  0.],
2075
           [ 0.,  0.]],
2076

2077
          [[ 0.,  0.],
2078
           [ 0.,  0.],
2079
           [ 0.,  0.],
2080
           [ 0.,  0.]],
2081

2082
          [[ 0.,  0.],
2083
           [ 0.,  0.],
2084
           [ 0.,  0.],
2085
           [ 0.,  0.]]],
2086

2087

2088
         [[[20., 30.],
2089
           [21., 31.],
2090
           [22., 32.],
2091
           [23., 33.]],
2092

2093
          [[21., 31.],
2094
           [22., 32.],
2095
           [23., 33.],
2096
           [24., 34.]],
2097

2098
          [[22., 32.],
2099
           [23., 33.],
2100
           [24., 34.],
2101
           [25., 35.]]]],
2102

2103

2104

2105
        [[[[ 2., 12.],
2106
           [ 3., 13.],
2107
           [ 4., 14.],
2108
           [ 5., 15.]],
2109

2110
          [[ 0.,  0.],
2111
           [ 0.,  0.],
2112
           [ 0.,  0.],
2113
           [ 0.,  0.]],
2114

2115
          [[ 3., 13.],
2116
           [ 4., 14.],
2117
           [ 5., 15.],
2118
           [ 6., 16.]]],
2119

2120

2121
         [[[ 0.,  0.],
2122
           [ 0.,  0.],
2123
           [ 0.,  0.],
2124
           [ 0.,  0.]],
2125

2126
          [[ 5., 15.],
2127
           [ 6., 16.],
2128
           [ 7., 17.],
2129
           [ 8., 18.]],
2130

2131
          [[ 0.,  0.],
2132
           [ 0.,  0.],
2133
           [ 0.,  0.],
2134
           [ 0.,  0.]]]],
2135

2136

2137

2138
        [[[[10., 20.],
2139
           [11., 21.],
2140
           [12., 22.],
2141
           [13., 23.]],
2142

2143
          [[11., 21.],
2144
           [12., 22.],
2145
           [13., 23.],
2146
           [14., 24.]],
2147

2148
          [[12., 22.],
2149
           [13., 23.],
2150
           [14., 24.],
2151
           [15., 25.]]],
2152

2153

2154
         [[[15., 25.],
2155
           [16., 26.],
2156
           [17., 27.],
2157
           [18., 28.]],
2158

2159
          [[16., 26.],
2160
           [17., 27.],
2161
           [18., 28.],
2162
           [19., 29.]],
2163

2164
          [[17., 27.],
2165
           [18., 28.],
2166
           [19., 29.],
2167
           [20., 30.]]]],
2168

2169

2170

2171
        [[[[ 0.,  0.],
2172
           [ 0.,  0.],
2173
           [ 0.,  0.],
2174
           [ 0.,  0.]],
2175

2176
          [[18., 28.],
2177
           [19., 29.],
2178
           [20., 30.],
2179
           [21., 31.]],
2180

2181
          [[19., 29.],
2182
           [20., 30.],
2183
           [21., 31.],
2184
           [22., 32.]]],
2185

2186

2187
         [[[ 0.,  0.],
2188
           [ 0.,  0.],
2189
           [ 0.,  0.],
2190
           [ 0.,  0.]],
2191

2192
          [[23., 33.],
2193
           [24., 34.],
2194
           [25., 35.],
2195
           [26., 36.]],
2196

2197
          [[24., 34.],
2198
           [25., 35.],
2199
           [26., 36.],
2200
           [27., 37.]]]]], device='cuda:0', dtype=torch.float64)
2201

2202

2203
########## torch.float32/torch.int64/size=(2, 3)+(2, 3)+(4,) ##########
2204
# sparse tensor
2205
tensor(ccol_indices=tensor([[[0, 1, 2, 3],
2206
                             [0, 1, 2, 3],
2207
                             [0, 1, 2, 3]],
2208

2209
                            [[0, 1, 2, 3],
2210
                             [0, 1, 2, 3],
2211
                             [0, 1, 2, 3]]]),
2212
       row_indices=tensor([[[0, 0, 0],
2213
                            [0, 0, 0],
2214
                            [0, 0, 0]],
2215

2216
                           [[0, 0, 0],
2217
                            [0, 0, 0],
2218
                            [0, 0, 0]]]),
2219
       values=tensor([[[[[[1., 2., 3., 4.]],
2220

2221
                         [[3., 4., 5., 6.]]],
2222

2223

2224
                        [[[2., 3., 4., 5.]],
2225

2226
                         [[0., 0., 0., 0.]]],
2227

2228

2229
                        [[[0., 0., 0., 0.]],
2230

2231
                         [[4., 5., 6., 7.]]]],
2232

2233

2234

2235
                       [[[[1., 2., 3., 4.]],
2236

2237
                         [[4., 5., 6., 7.]]],
2238

2239

2240
                        [[[2., 3., 4., 5.]],
2241

2242
                         [[0., 0., 0., 0.]]],
2243

2244

2245
                        [[[3., 4., 5., 6.]],
2246

2247
                         [[0., 0., 0., 0.]]]],
2248

2249

2250

2251
                       [[[[1., 2., 3., 4.]],
2252

2253
                         [[2., 3., 4., 5.]]],
2254

2255

2256
                        [[[0., 0., 0., 0.]],
2257

2258
                         [[3., 4., 5., 6.]]],
2259

2260

2261
                        [[[0., 0., 0., 0.]],
2262

2263
                         [[4., 5., 6., 7.]]]]],
2264

2265

2266

2267

2268
                      [[[[[0., 0., 0., 0.]],
2269

2270
                         [[2., 3., 4., 5.]]],
2271

2272

2273
                        [[[1., 2., 3., 4.]],
2274

2275
                         [[3., 4., 5., 6.]]],
2276

2277

2278
                        [[[0., 0., 0., 0.]],
2279

2280
                         [[4., 5., 6., 7.]]]],
2281

2282

2283

2284
                       [[[[1., 2., 3., 4.]],
2285

2286
                         [[3., 4., 5., 6.]]],
2287

2288

2289
                        [[[0., 0., 0., 0.]],
2290

2291
                         [[4., 5., 6., 7.]]],
2292

2293

2294
                        [[[2., 3., 4., 5.]],
2295

2296
                         [[0., 0., 0., 0.]]]],
2297

2298

2299

2300
                       [[[[1., 2., 3., 4.]],
2301

2302
                         [[0., 0., 0., 0.]]],
2303

2304

2305
                        [[[2., 3., 4., 5.]],
2306

2307
                         [[4., 5., 6., 7.]]],
2308

2309

2310
                        [[[3., 4., 5., 6.]],
2311

2312
                         [[0., 0., 0., 0.]]]]]]), device='cuda:0',
2313
       size=(2, 3, 2, 3, 4), nnz=3, layout=torch.sparse_bsc)
2314
# _ccol_indices
2315
tensor([[[0, 1, 2, 3],
2316
         [0, 1, 2, 3],
2317
         [0, 1, 2, 3]],
2318

2319
        [[0, 1, 2, 3],
2320
         [0, 1, 2, 3],
2321
         [0, 1, 2, 3]]], device='cuda:0')
2322
# _row_indices
2323
tensor([[[0, 0, 0],
2324
         [0, 0, 0],
2325
         [0, 0, 0]],
2326

2327
        [[0, 0, 0],
2328
         [0, 0, 0],
2329
         [0, 0, 0]]], device='cuda:0')
2330
# _values
2331
tensor([[[[[[1., 2., 3., 4.]],
2332

2333
           [[3., 4., 5., 6.]]],
2334

2335

2336
          [[[2., 3., 4., 5.]],
2337

2338
           [[0., 0., 0., 0.]]],
2339

2340

2341
          [[[0., 0., 0., 0.]],
2342

2343
           [[4., 5., 6., 7.]]]],
2344

2345

2346

2347
         [[[[1., 2., 3., 4.]],
2348

2349
           [[4., 5., 6., 7.]]],
2350

2351

2352
          [[[2., 3., 4., 5.]],
2353

2354
           [[0., 0., 0., 0.]]],
2355

2356

2357
          [[[3., 4., 5., 6.]],
2358

2359
           [[0., 0., 0., 0.]]]],
2360

2361

2362

2363
         [[[[1., 2., 3., 4.]],
2364

2365
           [[2., 3., 4., 5.]]],
2366

2367

2368
          [[[0., 0., 0., 0.]],
2369

2370
           [[3., 4., 5., 6.]]],
2371

2372

2373
          [[[0., 0., 0., 0.]],
2374

2375
           [[4., 5., 6., 7.]]]]],
2376

2377

2378

2379

2380
        [[[[[0., 0., 0., 0.]],
2381

2382
           [[2., 3., 4., 5.]]],
2383

2384

2385
          [[[1., 2., 3., 4.]],
2386

2387
           [[3., 4., 5., 6.]]],
2388

2389

2390
          [[[0., 0., 0., 0.]],
2391

2392
           [[4., 5., 6., 7.]]]],
2393

2394

2395

2396
         [[[[1., 2., 3., 4.]],
2397

2398
           [[3., 4., 5., 6.]]],
2399

2400

2401
          [[[0., 0., 0., 0.]],
2402

2403
           [[4., 5., 6., 7.]]],
2404

2405

2406
          [[[2., 3., 4., 5.]],
2407

2408
           [[0., 0., 0., 0.]]]],
2409

2410

2411

2412
         [[[[1., 2., 3., 4.]],
2413

2414
           [[0., 0., 0., 0.]]],
2415

2416

2417
          [[[2., 3., 4., 5.]],
2418

2419
           [[4., 5., 6., 7.]]],
2420

2421

2422
          [[[3., 4., 5., 6.]],
2423

2424
           [[0., 0., 0., 0.]]]]]], device='cuda:0')
2425

2426
########## torch.float32/torch.int64/size=()+(8, 6)+(4, 2) ##########
2427
# sparse tensor
2428
tensor(ccol_indices=tensor([0, 4, 7]),
2429
       row_indices=tensor([0, 1, 2, 3, 0, 2, 3]),
2430
       values=tensor([[[[[ 0.,  0.],
2431
                         [ 0.,  0.],
2432
                         [ 0.,  0.],
2433
                         [ 0.,  0.]],
2434

2435
                        [[ 1., 11.],
2436
                         [ 2., 12.],
2437
                         [ 3., 13.],
2438
                         [ 4., 14.]],
2439

2440
                        [[ 0.,  0.],
2441
                         [ 0.,  0.],
2442
                         [ 0.,  0.],
2443
                         [ 0.,  0.]]],
2444

2445

2446
                       [[[ 0.,  0.],
2447
                         [ 0.,  0.],
2448
                         [ 0.,  0.],
2449
                         [ 0.,  0.]],
2450

2451
                        [[ 4., 14.],
2452
                         [ 5., 15.],
2453
                         [ 6., 16.],
2454
                         [ 7., 17.]],
2455

2456
                        [[ 0.,  0.],
2457
                         [ 0.,  0.],
2458
                         [ 0.,  0.],
2459
                         [ 0.,  0.]]]],
2460

2461

2462

2463
                      [[[[ 6., 16.],
2464
                         [ 7., 17.],
2465
                         [ 8., 18.],
2466
                         [ 9., 19.]],
2467

2468
                        [[ 7., 17.],
2469
                         [ 8., 18.],
2470
                         [ 9., 19.],
2471
                         [10., 20.]],
2472

2473
                        [[ 8., 18.],
2474
                         [ 9., 19.],
2475
                         [10., 20.],
2476
                         [11., 21.]]],
2477

2478

2479
                       [[[ 0.,  0.],
2480
                         [ 0.,  0.],
2481
                         [ 0.,  0.],
2482
                         [ 0.,  0.]],
2483

2484
                        [[ 0.,  0.],
2485
                         [ 0.,  0.],
2486
                         [ 0.,  0.],
2487
                         [ 0.,  0.]],
2488

2489
                        [[ 0.,  0.],
2490
                         [ 0.,  0.],
2491
                         [ 0.,  0.],
2492
                         [ 0.,  0.]]]],
2493

2494

2495

2496
                      [[[[ 0.,  0.],
2497
                         [ 0.,  0.],
2498
                         [ 0.,  0.],
2499
                         [ 0.,  0.]],
2500

2501
                        [[ 9., 19.],
2502
                         [10., 20.],
2503
                         [11., 21.],
2504
                         [12., 22.]],
2505

2506
                        [[ 0.,  0.],
2507
                         [ 0.,  0.],
2508
                         [ 0.,  0.],
2509
                         [ 0.,  0.]]],
2510

2511

2512
                       [[[13., 23.],
2513
                         [14., 24.],
2514
                         [15., 25.],
2515
                         [16., 26.]],
2516

2517
                        [[ 0.,  0.],
2518
                         [ 0.,  0.],
2519
                         [ 0.,  0.],
2520
                         [ 0.,  0.]],
2521

2522
                        [[14., 24.],
2523
                         [15., 25.],
2524
                         [16., 26.],
2525
                         [17., 27.]]]],
2526

2527

2528

2529
                      [[[[ 0.,  0.],
2530
                         [ 0.,  0.],
2531
                         [ 0.,  0.],
2532
                         [ 0.,  0.]],
2533

2534
                        [[ 0.,  0.],
2535
                         [ 0.,  0.],
2536
                         [ 0.,  0.],
2537
                         [ 0.,  0.]],
2538

2539
                        [[ 0.,  0.],
2540
                         [ 0.,  0.],
2541
                         [ 0.,  0.],
2542
                         [ 0.,  0.]]],
2543

2544

2545
                       [[[20., 30.],
2546
                         [21., 31.],
2547
                         [22., 32.],
2548
                         [23., 33.]],
2549

2550
                        [[21., 31.],
2551
                         [22., 32.],
2552
                         [23., 33.],
2553
                         [24., 34.]],
2554

2555
                        [[22., 32.],
2556
                         [23., 33.],
2557
                         [24., 34.],
2558
                         [25., 35.]]]],
2559

2560

2561

2562
                      [[[[ 2., 12.],
2563
                         [ 3., 13.],
2564
                         [ 4., 14.],
2565
                         [ 5., 15.]],
2566

2567
                        [[ 0.,  0.],
2568
                         [ 0.,  0.],
2569
                         [ 0.,  0.],
2570
                         [ 0.,  0.]],
2571

2572
                        [[ 3., 13.],
2573
                         [ 4., 14.],
2574
                         [ 5., 15.],
2575
                         [ 6., 16.]]],
2576

2577

2578
                       [[[ 0.,  0.],
2579
                         [ 0.,  0.],
2580
                         [ 0.,  0.],
2581
                         [ 0.,  0.]],
2582

2583
                        [[ 5., 15.],
2584
                         [ 6., 16.],
2585
                         [ 7., 17.],
2586
                         [ 8., 18.]],
2587

2588
                        [[ 0.,  0.],
2589
                         [ 0.,  0.],
2590
                         [ 0.,  0.],
2591
                         [ 0.,  0.]]]],
2592

2593

2594

2595
                      [[[[10., 20.],
2596
                         [11., 21.],
2597
                         [12., 22.],
2598
                         [13., 23.]],
2599

2600
                        [[11., 21.],
2601
                         [12., 22.],
2602
                         [13., 23.],
2603
                         [14., 24.]],
2604

2605
                        [[12., 22.],
2606
                         [13., 23.],
2607
                         [14., 24.],
2608
                         [15., 25.]]],
2609

2610

2611
                       [[[15., 25.],
2612
                         [16., 26.],
2613
                         [17., 27.],
2614
                         [18., 28.]],
2615

2616
                        [[16., 26.],
2617
                         [17., 27.],
2618
                         [18., 28.],
2619
                         [19., 29.]],
2620

2621
                        [[17., 27.],
2622
                         [18., 28.],
2623
                         [19., 29.],
2624
                         [20., 30.]]]],
2625

2626

2627

2628
                      [[[[ 0.,  0.],
2629
                         [ 0.,  0.],
2630
                         [ 0.,  0.],
2631
                         [ 0.,  0.]],
2632

2633
                        [[18., 28.],
2634
                         [19., 29.],
2635
                         [20., 30.],
2636
                         [21., 31.]],
2637

2638
                        [[19., 29.],
2639
                         [20., 30.],
2640
                         [21., 31.],
2641
                         [22., 32.]]],
2642

2643

2644
                       [[[ 0.,  0.],
2645
                         [ 0.,  0.],
2646
                         [ 0.,  0.],
2647
                         [ 0.,  0.]],
2648

2649
                        [[23., 33.],
2650
                         [24., 34.],
2651
                         [25., 35.],
2652
                         [26., 36.]],
2653

2654
                        [[24., 34.],
2655
                         [25., 35.],
2656
                         [26., 36.],
2657
                         [27., 37.]]]]]), device='cuda:0', size=(8, 6, 4, 2),
2658
       nnz=7, layout=torch.sparse_bsc)
2659
# _ccol_indices
2660
tensor([0, 4, 7], device='cuda:0')
2661
# _row_indices
2662
tensor([0, 1, 2, 3, 0, 2, 3], device='cuda:0')
2663
# _values
2664
tensor([[[[[ 0.,  0.],
2665
           [ 0.,  0.],
2666
           [ 0.,  0.],
2667
           [ 0.,  0.]],
2668

2669
          [[ 1., 11.],
2670
           [ 2., 12.],
2671
           [ 3., 13.],
2672
           [ 4., 14.]],
2673

2674
          [[ 0.,  0.],
2675
           [ 0.,  0.],
2676
           [ 0.,  0.],
2677
           [ 0.,  0.]]],
2678

2679

2680
         [[[ 0.,  0.],
2681
           [ 0.,  0.],
2682
           [ 0.,  0.],
2683
           [ 0.,  0.]],
2684

2685
          [[ 4., 14.],
2686
           [ 5., 15.],
2687
           [ 6., 16.],
2688
           [ 7., 17.]],
2689

2690
          [[ 0.,  0.],
2691
           [ 0.,  0.],
2692
           [ 0.,  0.],
2693
           [ 0.,  0.]]]],
2694

2695

2696

2697
        [[[[ 6., 16.],
2698
           [ 7., 17.],
2699
           [ 8., 18.],
2700
           [ 9., 19.]],
2701

2702
          [[ 7., 17.],
2703
           [ 8., 18.],
2704
           [ 9., 19.],
2705
           [10., 20.]],
2706

2707
          [[ 8., 18.],
2708
           [ 9., 19.],
2709
           [10., 20.],
2710
           [11., 21.]]],
2711

2712

2713
         [[[ 0.,  0.],
2714
           [ 0.,  0.],
2715
           [ 0.,  0.],
2716
           [ 0.,  0.]],
2717

2718
          [[ 0.,  0.],
2719
           [ 0.,  0.],
2720
           [ 0.,  0.],
2721
           [ 0.,  0.]],
2722

2723
          [[ 0.,  0.],
2724
           [ 0.,  0.],
2725
           [ 0.,  0.],
2726
           [ 0.,  0.]]]],
2727

2728

2729

2730
        [[[[ 0.,  0.],
2731
           [ 0.,  0.],
2732
           [ 0.,  0.],
2733
           [ 0.,  0.]],
2734

2735
          [[ 9., 19.],
2736
           [10., 20.],
2737
           [11., 21.],
2738
           [12., 22.]],
2739

2740
          [[ 0.,  0.],
2741
           [ 0.,  0.],
2742
           [ 0.,  0.],
2743
           [ 0.,  0.]]],
2744

2745

2746
         [[[13., 23.],
2747
           [14., 24.],
2748
           [15., 25.],
2749
           [16., 26.]],
2750

2751
          [[ 0.,  0.],
2752
           [ 0.,  0.],
2753
           [ 0.,  0.],
2754
           [ 0.,  0.]],
2755

2756
          [[14., 24.],
2757
           [15., 25.],
2758
           [16., 26.],
2759
           [17., 27.]]]],
2760

2761

2762

2763
        [[[[ 0.,  0.],
2764
           [ 0.,  0.],
2765
           [ 0.,  0.],
2766
           [ 0.,  0.]],
2767

2768
          [[ 0.,  0.],
2769
           [ 0.,  0.],
2770
           [ 0.,  0.],
2771
           [ 0.,  0.]],
2772

2773
          [[ 0.,  0.],
2774
           [ 0.,  0.],
2775
           [ 0.,  0.],
2776
           [ 0.,  0.]]],
2777

2778

2779
         [[[20., 30.],
2780
           [21., 31.],
2781
           [22., 32.],
2782
           [23., 33.]],
2783

2784
          [[21., 31.],
2785
           [22., 32.],
2786
           [23., 33.],
2787
           [24., 34.]],
2788

2789
          [[22., 32.],
2790
           [23., 33.],
2791
           [24., 34.],
2792
           [25., 35.]]]],
2793

2794

2795

2796
        [[[[ 2., 12.],
2797
           [ 3., 13.],
2798
           [ 4., 14.],
2799
           [ 5., 15.]],
2800

2801
          [[ 0.,  0.],
2802
           [ 0.,  0.],
2803
           [ 0.,  0.],
2804
           [ 0.,  0.]],
2805

2806
          [[ 3., 13.],
2807
           [ 4., 14.],
2808
           [ 5., 15.],
2809
           [ 6., 16.]]],
2810

2811

2812
         [[[ 0.,  0.],
2813
           [ 0.,  0.],
2814
           [ 0.,  0.],
2815
           [ 0.,  0.]],
2816

2817
          [[ 5., 15.],
2818
           [ 6., 16.],
2819
           [ 7., 17.],
2820
           [ 8., 18.]],
2821

2822
          [[ 0.,  0.],
2823
           [ 0.,  0.],
2824
           [ 0.,  0.],
2825
           [ 0.,  0.]]]],
2826

2827

2828

2829
        [[[[10., 20.],
2830
           [11., 21.],
2831
           [12., 22.],
2832
           [13., 23.]],
2833

2834
          [[11., 21.],
2835
           [12., 22.],
2836
           [13., 23.],
2837
           [14., 24.]],
2838

2839
          [[12., 22.],
2840
           [13., 23.],
2841
           [14., 24.],
2842
           [15., 25.]]],
2843

2844

2845
         [[[15., 25.],
2846
           [16., 26.],
2847
           [17., 27.],
2848
           [18., 28.]],
2849

2850
          [[16., 26.],
2851
           [17., 27.],
2852
           [18., 28.],
2853
           [19., 29.]],
2854

2855
          [[17., 27.],
2856
           [18., 28.],
2857
           [19., 29.],
2858
           [20., 30.]]]],
2859

2860

2861

2862
        [[[[ 0.,  0.],
2863
           [ 0.,  0.],
2864
           [ 0.,  0.],
2865
           [ 0.,  0.]],
2866

2867
          [[18., 28.],
2868
           [19., 29.],
2869
           [20., 30.],
2870
           [21., 31.]],
2871

2872
          [[19., 29.],
2873
           [20., 30.],
2874
           [21., 31.],
2875
           [22., 32.]]],
2876

2877

2878
         [[[ 0.,  0.],
2879
           [ 0.,  0.],
2880
           [ 0.,  0.],
2881
           [ 0.,  0.]],
2882

2883
          [[23., 33.],
2884
           [24., 34.],
2885
           [25., 35.],
2886
           [26., 36.]],
2887

2888
          [[24., 34.],
2889
           [25., 35.],
2890
           [26., 36.],
2891
           [27., 37.]]]]], device='cuda:0')
2892

2893

2894
########## torch.float64/torch.int64/size=(2, 3)+(2, 3)+(4,) ##########
2895
# sparse tensor
2896
tensor(ccol_indices=tensor([[[0, 1, 2, 3],
2897
                             [0, 1, 2, 3],
2898
                             [0, 1, 2, 3]],
2899

2900
                            [[0, 1, 2, 3],
2901
                             [0, 1, 2, 3],
2902
                             [0, 1, 2, 3]]]),
2903
       row_indices=tensor([[[0, 0, 0],
2904
                            [0, 0, 0],
2905
                            [0, 0, 0]],
2906

2907
                           [[0, 0, 0],
2908
                            [0, 0, 0],
2909
                            [0, 0, 0]]]),
2910
       values=tensor([[[[[[1., 2., 3., 4.]],
2911

2912
                         [[3., 4., 5., 6.]]],
2913

2914

2915
                        [[[2., 3., 4., 5.]],
2916

2917
                         [[0., 0., 0., 0.]]],
2918

2919

2920
                        [[[0., 0., 0., 0.]],
2921

2922
                         [[4., 5., 6., 7.]]]],
2923

2924

2925

2926
                       [[[[1., 2., 3., 4.]],
2927

2928
                         [[4., 5., 6., 7.]]],
2929

2930

2931
                        [[[2., 3., 4., 5.]],
2932

2933
                         [[0., 0., 0., 0.]]],
2934

2935

2936
                        [[[3., 4., 5., 6.]],
2937

2938
                         [[0., 0., 0., 0.]]]],
2939

2940

2941

2942
                       [[[[1., 2., 3., 4.]],
2943

2944
                         [[2., 3., 4., 5.]]],
2945

2946

2947
                        [[[0., 0., 0., 0.]],
2948

2949
                         [[3., 4., 5., 6.]]],
2950

2951

2952
                        [[[0., 0., 0., 0.]],
2953

2954
                         [[4., 5., 6., 7.]]]]],
2955

2956

2957

2958

2959
                      [[[[[0., 0., 0., 0.]],
2960

2961
                         [[2., 3., 4., 5.]]],
2962

2963

2964
                        [[[1., 2., 3., 4.]],
2965

2966
                         [[3., 4., 5., 6.]]],
2967

2968

2969
                        [[[0., 0., 0., 0.]],
2970

2971
                         [[4., 5., 6., 7.]]]],
2972

2973

2974

2975
                       [[[[1., 2., 3., 4.]],
2976

2977
                         [[3., 4., 5., 6.]]],
2978

2979

2980
                        [[[0., 0., 0., 0.]],
2981

2982
                         [[4., 5., 6., 7.]]],
2983

2984

2985
                        [[[2., 3., 4., 5.]],
2986

2987
                         [[0., 0., 0., 0.]]]],
2988

2989

2990

2991
                       [[[[1., 2., 3., 4.]],
2992

2993
                         [[0., 0., 0., 0.]]],
2994

2995

2996
                        [[[2., 3., 4., 5.]],
2997

2998
                         [[4., 5., 6., 7.]]],
2999

3000

3001
                        [[[3., 4., 5., 6.]],
3002

3003
                         [[0., 0., 0., 0.]]]]]]), device='cuda:0',
3004
       size=(2, 3, 2, 3, 4), nnz=3, dtype=torch.float64, layout=torch.sparse_bsc)
3005
# _ccol_indices
3006
tensor([[[0, 1, 2, 3],
3007
         [0, 1, 2, 3],
3008
         [0, 1, 2, 3]],
3009

3010
        [[0, 1, 2, 3],
3011
         [0, 1, 2, 3],
3012
         [0, 1, 2, 3]]], device='cuda:0')
3013
# _row_indices
3014
tensor([[[0, 0, 0],
3015
         [0, 0, 0],
3016
         [0, 0, 0]],
3017

3018
        [[0, 0, 0],
3019
         [0, 0, 0],
3020
         [0, 0, 0]]], device='cuda:0')
3021
# _values
3022
tensor([[[[[[1., 2., 3., 4.]],
3023

3024
           [[3., 4., 5., 6.]]],
3025

3026

3027
          [[[2., 3., 4., 5.]],
3028

3029
           [[0., 0., 0., 0.]]],
3030

3031

3032
          [[[0., 0., 0., 0.]],
3033

3034
           [[4., 5., 6., 7.]]]],
3035

3036

3037

3038
         [[[[1., 2., 3., 4.]],
3039

3040
           [[4., 5., 6., 7.]]],
3041

3042

3043
          [[[2., 3., 4., 5.]],
3044

3045
           [[0., 0., 0., 0.]]],
3046

3047

3048
          [[[3., 4., 5., 6.]],
3049

3050
           [[0., 0., 0., 0.]]]],
3051

3052

3053

3054
         [[[[1., 2., 3., 4.]],
3055

3056
           [[2., 3., 4., 5.]]],
3057

3058

3059
          [[[0., 0., 0., 0.]],
3060

3061
           [[3., 4., 5., 6.]]],
3062

3063

3064
          [[[0., 0., 0., 0.]],
3065

3066
           [[4., 5., 6., 7.]]]]],
3067

3068

3069

3070

3071
        [[[[[0., 0., 0., 0.]],
3072

3073
           [[2., 3., 4., 5.]]],
3074

3075

3076
          [[[1., 2., 3., 4.]],
3077

3078
           [[3., 4., 5., 6.]]],
3079

3080

3081
          [[[0., 0., 0., 0.]],
3082

3083
           [[4., 5., 6., 7.]]]],
3084

3085

3086

3087
         [[[[1., 2., 3., 4.]],
3088

3089
           [[3., 4., 5., 6.]]],
3090

3091

3092
          [[[0., 0., 0., 0.]],
3093

3094
           [[4., 5., 6., 7.]]],
3095

3096

3097
          [[[2., 3., 4., 5.]],
3098

3099
           [[0., 0., 0., 0.]]]],
3100

3101

3102

3103
         [[[[1., 2., 3., 4.]],
3104

3105
           [[0., 0., 0., 0.]]],
3106

3107

3108
          [[[2., 3., 4., 5.]],
3109

3110
           [[4., 5., 6., 7.]]],
3111

3112

3113
          [[[3., 4., 5., 6.]],
3114

3115
           [[0., 0., 0., 0.]]]]]], device='cuda:0', dtype=torch.float64)
3116

3117
########## torch.float64/torch.int64/size=()+(8, 6)+(4, 2) ##########
3118
# sparse tensor
3119
tensor(ccol_indices=tensor([0, 4, 7]),
3120
       row_indices=tensor([0, 1, 2, 3, 0, 2, 3]),
3121
       values=tensor([[[[[ 0.,  0.],
3122
                         [ 0.,  0.],
3123
                         [ 0.,  0.],
3124
                         [ 0.,  0.]],
3125

3126
                        [[ 1., 11.],
3127
                         [ 2., 12.],
3128
                         [ 3., 13.],
3129
                         [ 4., 14.]],
3130

3131
                        [[ 0.,  0.],
3132
                         [ 0.,  0.],
3133
                         [ 0.,  0.],
3134
                         [ 0.,  0.]]],
3135

3136

3137
                       [[[ 0.,  0.],
3138
                         [ 0.,  0.],
3139
                         [ 0.,  0.],
3140
                         [ 0.,  0.]],
3141

3142
                        [[ 4., 14.],
3143
                         [ 5., 15.],
3144
                         [ 6., 16.],
3145
                         [ 7., 17.]],
3146

3147
                        [[ 0.,  0.],
3148
                         [ 0.,  0.],
3149
                         [ 0.,  0.],
3150
                         [ 0.,  0.]]]],
3151

3152

3153

3154
                      [[[[ 6., 16.],
3155
                         [ 7., 17.],
3156
                         [ 8., 18.],
3157
                         [ 9., 19.]],
3158

3159
                        [[ 7., 17.],
3160
                         [ 8., 18.],
3161
                         [ 9., 19.],
3162
                         [10., 20.]],
3163

3164
                        [[ 8., 18.],
3165
                         [ 9., 19.],
3166
                         [10., 20.],
3167
                         [11., 21.]]],
3168

3169

3170
                       [[[ 0.,  0.],
3171
                         [ 0.,  0.],
3172
                         [ 0.,  0.],
3173
                         [ 0.,  0.]],
3174

3175
                        [[ 0.,  0.],
3176
                         [ 0.,  0.],
3177
                         [ 0.,  0.],
3178
                         [ 0.,  0.]],
3179

3180
                        [[ 0.,  0.],
3181
                         [ 0.,  0.],
3182
                         [ 0.,  0.],
3183
                         [ 0.,  0.]]]],
3184

3185

3186

3187
                      [[[[ 0.,  0.],
3188
                         [ 0.,  0.],
3189
                         [ 0.,  0.],
3190
                         [ 0.,  0.]],
3191

3192
                        [[ 9., 19.],
3193
                         [10., 20.],
3194
                         [11., 21.],
3195
                         [12., 22.]],
3196

3197
                        [[ 0.,  0.],
3198
                         [ 0.,  0.],
3199
                         [ 0.,  0.],
3200
                         [ 0.,  0.]]],
3201

3202

3203
                       [[[13., 23.],
3204
                         [14., 24.],
3205
                         [15., 25.],
3206
                         [16., 26.]],
3207

3208
                        [[ 0.,  0.],
3209
                         [ 0.,  0.],
3210
                         [ 0.,  0.],
3211
                         [ 0.,  0.]],
3212

3213
                        [[14., 24.],
3214
                         [15., 25.],
3215
                         [16., 26.],
3216
                         [17., 27.]]]],
3217

3218

3219

3220
                      [[[[ 0.,  0.],
3221
                         [ 0.,  0.],
3222
                         [ 0.,  0.],
3223
                         [ 0.,  0.]],
3224

3225
                        [[ 0.,  0.],
3226
                         [ 0.,  0.],
3227
                         [ 0.,  0.],
3228
                         [ 0.,  0.]],
3229

3230
                        [[ 0.,  0.],
3231
                         [ 0.,  0.],
3232
                         [ 0.,  0.],
3233
                         [ 0.,  0.]]],
3234

3235

3236
                       [[[20., 30.],
3237
                         [21., 31.],
3238
                         [22., 32.],
3239
                         [23., 33.]],
3240

3241
                        [[21., 31.],
3242
                         [22., 32.],
3243
                         [23., 33.],
3244
                         [24., 34.]],
3245

3246
                        [[22., 32.],
3247
                         [23., 33.],
3248
                         [24., 34.],
3249
                         [25., 35.]]]],
3250

3251

3252

3253
                      [[[[ 2., 12.],
3254
                         [ 3., 13.],
3255
                         [ 4., 14.],
3256
                         [ 5., 15.]],
3257

3258
                        [[ 0.,  0.],
3259
                         [ 0.,  0.],
3260
                         [ 0.,  0.],
3261
                         [ 0.,  0.]],
3262

3263
                        [[ 3., 13.],
3264
                         [ 4., 14.],
3265
                         [ 5., 15.],
3266
                         [ 6., 16.]]],
3267

3268

3269
                       [[[ 0.,  0.],
3270
                         [ 0.,  0.],
3271
                         [ 0.,  0.],
3272
                         [ 0.,  0.]],
3273

3274
                        [[ 5., 15.],
3275
                         [ 6., 16.],
3276
                         [ 7., 17.],
3277
                         [ 8., 18.]],
3278

3279
                        [[ 0.,  0.],
3280
                         [ 0.,  0.],
3281
                         [ 0.,  0.],
3282
                         [ 0.,  0.]]]],
3283

3284

3285

3286
                      [[[[10., 20.],
3287
                         [11., 21.],
3288
                         [12., 22.],
3289
                         [13., 23.]],
3290

3291
                        [[11., 21.],
3292
                         [12., 22.],
3293
                         [13., 23.],
3294
                         [14., 24.]],
3295

3296
                        [[12., 22.],
3297
                         [13., 23.],
3298
                         [14., 24.],
3299
                         [15., 25.]]],
3300

3301

3302
                       [[[15., 25.],
3303
                         [16., 26.],
3304
                         [17., 27.],
3305
                         [18., 28.]],
3306

3307
                        [[16., 26.],
3308
                         [17., 27.],
3309
                         [18., 28.],
3310
                         [19., 29.]],
3311

3312
                        [[17., 27.],
3313
                         [18., 28.],
3314
                         [19., 29.],
3315
                         [20., 30.]]]],
3316

3317

3318

3319
                      [[[[ 0.,  0.],
3320
                         [ 0.,  0.],
3321
                         [ 0.,  0.],
3322
                         [ 0.,  0.]],
3323

3324
                        [[18., 28.],
3325
                         [19., 29.],
3326
                         [20., 30.],
3327
                         [21., 31.]],
3328

3329
                        [[19., 29.],
3330
                         [20., 30.],
3331
                         [21., 31.],
3332
                         [22., 32.]]],
3333

3334

3335
                       [[[ 0.,  0.],
3336
                         [ 0.,  0.],
3337
                         [ 0.,  0.],
3338
                         [ 0.,  0.]],
3339

3340
                        [[23., 33.],
3341
                         [24., 34.],
3342
                         [25., 35.],
3343
                         [26., 36.]],
3344

3345
                        [[24., 34.],
3346
                         [25., 35.],
3347
                         [26., 36.],
3348
                         [27., 37.]]]]]), device='cuda:0', size=(8, 6, 4, 2),
3349
       nnz=7, dtype=torch.float64, layout=torch.sparse_bsc)
3350
# _ccol_indices
3351
tensor([0, 4, 7], device='cuda:0')
3352
# _row_indices
3353
tensor([0, 1, 2, 3, 0, 2, 3], device='cuda:0')
3354
# _values
3355
tensor([[[[[ 0.,  0.],
3356
           [ 0.,  0.],
3357
           [ 0.,  0.],
3358
           [ 0.,  0.]],
3359

3360
          [[ 1., 11.],
3361
           [ 2., 12.],
3362
           [ 3., 13.],
3363
           [ 4., 14.]],
3364

3365
          [[ 0.,  0.],
3366
           [ 0.,  0.],
3367
           [ 0.,  0.],
3368
           [ 0.,  0.]]],
3369

3370

3371
         [[[ 0.,  0.],
3372
           [ 0.,  0.],
3373
           [ 0.,  0.],
3374
           [ 0.,  0.]],
3375

3376
          [[ 4., 14.],
3377
           [ 5., 15.],
3378
           [ 6., 16.],
3379
           [ 7., 17.]],
3380

3381
          [[ 0.,  0.],
3382
           [ 0.,  0.],
3383
           [ 0.,  0.],
3384
           [ 0.,  0.]]]],
3385

3386

3387

3388
        [[[[ 6., 16.],
3389
           [ 7., 17.],
3390
           [ 8., 18.],
3391
           [ 9., 19.]],
3392

3393
          [[ 7., 17.],
3394
           [ 8., 18.],
3395
           [ 9., 19.],
3396
           [10., 20.]],
3397

3398
          [[ 8., 18.],
3399
           [ 9., 19.],
3400
           [10., 20.],
3401
           [11., 21.]]],
3402

3403

3404
         [[[ 0.,  0.],
3405
           [ 0.,  0.],
3406
           [ 0.,  0.],
3407
           [ 0.,  0.]],
3408

3409
          [[ 0.,  0.],
3410
           [ 0.,  0.],
3411
           [ 0.,  0.],
3412
           [ 0.,  0.]],
3413

3414
          [[ 0.,  0.],
3415
           [ 0.,  0.],
3416
           [ 0.,  0.],
3417
           [ 0.,  0.]]]],
3418

3419

3420

3421
        [[[[ 0.,  0.],
3422
           [ 0.,  0.],
3423
           [ 0.,  0.],
3424
           [ 0.,  0.]],
3425

3426
          [[ 9., 19.],
3427
           [10., 20.],
3428
           [11., 21.],
3429
           [12., 22.]],
3430

3431
          [[ 0.,  0.],
3432
           [ 0.,  0.],
3433
           [ 0.,  0.],
3434
           [ 0.,  0.]]],
3435

3436

3437
         [[[13., 23.],
3438
           [14., 24.],
3439
           [15., 25.],
3440
           [16., 26.]],
3441

3442
          [[ 0.,  0.],
3443
           [ 0.,  0.],
3444
           [ 0.,  0.],
3445
           [ 0.,  0.]],
3446

3447
          [[14., 24.],
3448
           [15., 25.],
3449
           [16., 26.],
3450
           [17., 27.]]]],
3451

3452

3453

3454
        [[[[ 0.,  0.],
3455
           [ 0.,  0.],
3456
           [ 0.,  0.],
3457
           [ 0.,  0.]],
3458

3459
          [[ 0.,  0.],
3460
           [ 0.,  0.],
3461
           [ 0.,  0.],
3462
           [ 0.,  0.]],
3463

3464
          [[ 0.,  0.],
3465
           [ 0.,  0.],
3466
           [ 0.,  0.],
3467
           [ 0.,  0.]]],
3468

3469

3470
         [[[20., 30.],
3471
           [21., 31.],
3472
           [22., 32.],
3473
           [23., 33.]],
3474

3475
          [[21., 31.],
3476
           [22., 32.],
3477
           [23., 33.],
3478
           [24., 34.]],
3479

3480
          [[22., 32.],
3481
           [23., 33.],
3482
           [24., 34.],
3483
           [25., 35.]]]],
3484

3485

3486

3487
        [[[[ 2., 12.],
3488
           [ 3., 13.],
3489
           [ 4., 14.],
3490
           [ 5., 15.]],
3491

3492
          [[ 0.,  0.],
3493
           [ 0.,  0.],
3494
           [ 0.,  0.],
3495
           [ 0.,  0.]],
3496

3497
          [[ 3., 13.],
3498
           [ 4., 14.],
3499
           [ 5., 15.],
3500
           [ 6., 16.]]],
3501

3502

3503
         [[[ 0.,  0.],
3504
           [ 0.,  0.],
3505
           [ 0.,  0.],
3506
           [ 0.,  0.]],
3507

3508
          [[ 5., 15.],
3509
           [ 6., 16.],
3510
           [ 7., 17.],
3511
           [ 8., 18.]],
3512

3513
          [[ 0.,  0.],
3514
           [ 0.,  0.],
3515
           [ 0.,  0.],
3516
           [ 0.,  0.]]]],
3517

3518

3519

3520
        [[[[10., 20.],
3521
           [11., 21.],
3522
           [12., 22.],
3523
           [13., 23.]],
3524

3525
          [[11., 21.],
3526
           [12., 22.],
3527
           [13., 23.],
3528
           [14., 24.]],
3529

3530
          [[12., 22.],
3531
           [13., 23.],
3532
           [14., 24.],
3533
           [15., 25.]]],
3534

3535

3536
         [[[15., 25.],
3537
           [16., 26.],
3538
           [17., 27.],
3539
           [18., 28.]],
3540

3541
          [[16., 26.],
3542
           [17., 27.],
3543
           [18., 28.],
3544
           [19., 29.]],
3545

3546
          [[17., 27.],
3547
           [18., 28.],
3548
           [19., 29.],
3549
           [20., 30.]]]],
3550

3551

3552

3553
        [[[[ 0.,  0.],
3554
           [ 0.,  0.],
3555
           [ 0.,  0.],
3556
           [ 0.,  0.]],
3557

3558
          [[18., 28.],
3559
           [19., 29.],
3560
           [20., 30.],
3561
           [21., 31.]],
3562

3563
          [[19., 29.],
3564
           [20., 30.],
3565
           [21., 31.],
3566
           [22., 32.]]],
3567

3568

3569
         [[[ 0.,  0.],
3570
           [ 0.,  0.],
3571
           [ 0.,  0.],
3572
           [ 0.,  0.]],
3573

3574
          [[23., 33.],
3575
           [24., 34.],
3576
           [25., 35.],
3577
           [26., 36.]],
3578

3579
          [[24., 34.],
3580
           [25., 35.],
3581
           [26., 36.],
3582
           [27., 37.]]]]], device='cuda:0', dtype=torch.float64)
3583

3584

Использование cookies

Мы используем файлы cookie в соответствии с Политикой конфиденциальности и Политикой использования cookies.

Нажимая кнопку «Принимаю», Вы даете АО «СберТех» согласие на обработку Ваших персональных данных в целях совершенствования нашего веб-сайта и Сервиса GitVerse, а также повышения удобства их использования.

Запретить использование cookies Вы можете самостоятельно в настройках Вашего браузера.