Google Git
Sign in
android / platform / external / pytorch / d0e12d1cc8b08ea8770b6ec941372793c4e4d4d0 / . / test / expect / TestSparseCompressedCPU.test_print_SparseBSR_cpu.expect
blob: 267056b76e6786b66faa6b256c1a0634c36ab853 [file] [log] [blame]
########## torch.float32/torch.int32/size=()+(4, 3)+() ##########
# sparse tensor
tensor(crow_indices=tensor([0, 2, 4]),
col_indices=tensor([0, 1, 0, 2]),
values=tensor([[[1.],
[2.]],
[[2.],
[3.]],
[[3.],
[4.]],
[[4.],
[5.]]]), size=(4, 3), nnz=4, layout=torch.sparse_bsr)
# _crow_indices
tensor([0, 2, 4], dtype=torch.int32)
# _col_indices
tensor([0, 1, 0, 2], dtype=torch.int32)
# _values
tensor([[[1.],
[2.]],
[[2.],
[3.]],
[[3.],
[4.]],
[[4.],
[5.]]])
########## torch.float32/torch.int32/size=()+(0, 0)+() ##########
# sparse tensor
tensor(crow_indices=tensor([0]),
col_indices=tensor([], size=(0,)),
values=tensor([], size=(0, 2, 1)), size=(0, 0), nnz=0,
layout=torch.sparse_bsr)
# _crow_indices
tensor([0], dtype=torch.int32)
# _col_indices
tensor([], dtype=torch.int32)
# _values
tensor([], size=(0, 2, 1))
########## torch.float32/torch.int32/size=(2,)+(2, 6)+() ##########
# sparse tensor
tensor(crow_indices=tensor([[0, 2, 4],
[0, 3, 4]]),
col_indices=tensor([[0, 1, 0, 1],
[0, 1, 2, 0]]),
values=tensor([[[[ 1., 11.]],
[[ 2., 12.]],
[[ 3., 13.]],
[[ 4., 14.]]],
[[[ 5., 15.]],
[[ 6., 16.]],
[[ 7., 17.]],
[[ 8., 18.]]]]), size=(2, 2, 6), nnz=4,
layout=torch.sparse_bsr)
# _crow_indices
tensor([[0, 2, 4],
[0, 3, 4]], dtype=torch.int32)
# _col_indices
tensor([[0, 1, 0, 1],
[0, 1, 2, 0]], dtype=torch.int32)
# _values
tensor([[[[ 1., 11.]],
[[ 2., 12.]],
[[ 3., 13.]],
[[ 4., 14.]]],
[[[ 5., 15.]],
[[ 6., 16.]],
[[ 7., 17.]],
[[ 8., 18.]]]])
########## torch.float32/torch.int32/size=(2, 3)+(4, 9)+() ##########
# sparse tensor
tensor(crow_indices=tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]]),
col_indices=tensor([[[0, 1, 0, 1],
[0, 1, 2, 0],
[0, 0, 1, 2]],
[[1, 0, 1, 2],
[0, 2, 0, 1],
[0, 1, 2, 1]]]),
values=tensor([[[[[ 1., 11., 21.],
[ 2., 12., 22.]],
[[ 2., 12., 22.],
[ 3., 13., 23.]],
[[ 3., 13., 23.],
[ 4., 14., 24.]],
[[ 4., 14., 24.],
[ 5., 15., 25.]]],
[[[ 5., 15., 25.],
[ 6., 16., 26.]],
[[ 6., 16., 26.],
[ 7., 17., 27.]],
[[ 7., 17., 27.],
[ 8., 18., 28.]],
[[ 8., 18., 28.],
[ 9., 19., 29.]]],
[[[ 9., 19., 29.],
[10., 20., 30.]],
[[10., 20., 30.],
[11., 21., 31.]],
[[11., 21., 31.],
[12., 22., 32.]],
[[12., 22., 32.],
[13., 23., 33.]]]],
[[[[13., 23., 33.],
[14., 24., 34.]],
[[14., 24., 34.],
[15., 25., 35.]],
[[15., 25., 35.],
[16., 26., 36.]],
[[16., 26., 36.],
[17., 27., 37.]]],
[[[17., 27., 37.],
[18., 28., 38.]],
[[18., 28., 38.],
[19., 29., 39.]],
[[19., 29., 39.],
[20., 30., 40.]],
[[20., 30., 40.],
[21., 31., 41.]]],
[[[21., 31., 41.],
[22., 32., 42.]],
[[22., 32., 42.],
[23., 33., 43.]],
[[23., 33., 43.],
[24., 34., 44.]],
[[24., 34., 44.],
[25., 35., 45.]]]]]), size=(2, 3, 4, 9), nnz=4,
layout=torch.sparse_bsr)
# _crow_indices
tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]], dtype=torch.int32)
# _col_indices
tensor([[[0, 1, 0, 1],
[0, 1, 2, 0],
[0, 0, 1, 2]],
[[1, 0, 1, 2],
[0, 2, 0, 1],
[0, 1, 2, 1]]], dtype=torch.int32)
# _values
tensor([[[[[ 1., 11., 21.],
[ 2., 12., 22.]],
[[ 2., 12., 22.],
[ 3., 13., 23.]],
[[ 3., 13., 23.],
[ 4., 14., 24.]],
[[ 4., 14., 24.],
[ 5., 15., 25.]]],
[[[ 5., 15., 25.],
[ 6., 16., 26.]],
[[ 6., 16., 26.],
[ 7., 17., 27.]],
[[ 7., 17., 27.],
[ 8., 18., 28.]],
[[ 8., 18., 28.],
[ 9., 19., 29.]]],
[[[ 9., 19., 29.],
[10., 20., 30.]],
[[10., 20., 30.],
[11., 21., 31.]],
[[11., 21., 31.],
[12., 22., 32.]],
[[12., 22., 32.],
[13., 23., 33.]]]],
[[[[13., 23., 33.],
[14., 24., 34.]],
[[14., 24., 34.],
[15., 25., 35.]],
[[15., 25., 35.],
[16., 26., 36.]],
[[16., 26., 36.],
[17., 27., 37.]]],
[[[17., 27., 37.],
[18., 28., 38.]],
[[18., 28., 38.],
[19., 29., 39.]],
[[19., 29., 39.],
[20., 30., 40.]],
[[20., 30., 40.],
[21., 31., 41.]]],
[[[21., 31., 41.],
[22., 32., 42.]],
[[22., 32., 42.],
[23., 33., 43.]],
[[23., 33., 43.],
[24., 34., 44.]],
[[24., 34., 44.],
[25., 35., 45.]]]]])
########## torch.float64/torch.int32/size=()+(4, 3)+() ##########
# sparse tensor
tensor(crow_indices=tensor([0, 2, 4]),
col_indices=tensor([0, 1, 0, 2]),
values=tensor([[[1.],
[2.]],
[[2.],
[3.]],
[[3.],
[4.]],
[[4.],
[5.]]]), size=(4, 3), nnz=4, dtype=torch.float64,
layout=torch.sparse_bsr)
# _crow_indices
tensor([0, 2, 4], dtype=torch.int32)
# _col_indices
tensor([0, 1, 0, 2], dtype=torch.int32)
# _values
tensor([[[1.],
[2.]],
[[2.],
[3.]],
[[3.],
[4.]],
[[4.],
[5.]]], dtype=torch.float64)
########## torch.float64/torch.int32/size=()+(0, 0)+() ##########
# sparse tensor
tensor(crow_indices=tensor([0]),
col_indices=tensor([], size=(0,)),
values=tensor([], size=(0, 2, 1)), size=(0, 0), nnz=0,
dtype=torch.float64, layout=torch.sparse_bsr)
# _crow_indices
tensor([0], dtype=torch.int32)
# _col_indices
tensor([], dtype=torch.int32)
# _values
tensor([], size=(0, 2, 1), dtype=torch.float64)
########## torch.float64/torch.int32/size=(2,)+(2, 6)+() ##########
# sparse tensor
tensor(crow_indices=tensor([[0, 2, 4],
[0, 3, 4]]),
col_indices=tensor([[0, 1, 0, 1],
[0, 1, 2, 0]]),
values=tensor([[[[ 1., 11.]],
[[ 2., 12.]],
[[ 3., 13.]],
[[ 4., 14.]]],
[[[ 5., 15.]],
[[ 6., 16.]],
[[ 7., 17.]],
[[ 8., 18.]]]]), size=(2, 2, 6), nnz=4,
dtype=torch.float64, layout=torch.sparse_bsr)
# _crow_indices
tensor([[0, 2, 4],
[0, 3, 4]], dtype=torch.int32)
# _col_indices
tensor([[0, 1, 0, 1],
[0, 1, 2, 0]], dtype=torch.int32)
# _values
tensor([[[[ 1., 11.]],
[[ 2., 12.]],
[[ 3., 13.]],
[[ 4., 14.]]],
[[[ 5., 15.]],
[[ 6., 16.]],
[[ 7., 17.]],
[[ 8., 18.]]]], dtype=torch.float64)
########## torch.float64/torch.int32/size=(2, 3)+(4, 9)+() ##########
# sparse tensor
tensor(crow_indices=tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]]),
col_indices=tensor([[[0, 1, 0, 1],
[0, 1, 2, 0],
[0, 0, 1, 2]],
[[1, 0, 1, 2],
[0, 2, 0, 1],
[0, 1, 2, 1]]]),
values=tensor([[[[[ 1., 11., 21.],
[ 2., 12., 22.]],
[[ 2., 12., 22.],
[ 3., 13., 23.]],
[[ 3., 13., 23.],
[ 4., 14., 24.]],
[[ 4., 14., 24.],
[ 5., 15., 25.]]],
[[[ 5., 15., 25.],
[ 6., 16., 26.]],
[[ 6., 16., 26.],
[ 7., 17., 27.]],
[[ 7., 17., 27.],
[ 8., 18., 28.]],
[[ 8., 18., 28.],
[ 9., 19., 29.]]],
[[[ 9., 19., 29.],
[10., 20., 30.]],
[[10., 20., 30.],
[11., 21., 31.]],
[[11., 21., 31.],
[12., 22., 32.]],
[[12., 22., 32.],
[13., 23., 33.]]]],
[[[[13., 23., 33.],
[14., 24., 34.]],
[[14., 24., 34.],
[15., 25., 35.]],
[[15., 25., 35.],
[16., 26., 36.]],
[[16., 26., 36.],
[17., 27., 37.]]],
[[[17., 27., 37.],
[18., 28., 38.]],
[[18., 28., 38.],
[19., 29., 39.]],
[[19., 29., 39.],
[20., 30., 40.]],
[[20., 30., 40.],
[21., 31., 41.]]],
[[[21., 31., 41.],
[22., 32., 42.]],
[[22., 32., 42.],
[23., 33., 43.]],
[[23., 33., 43.],
[24., 34., 44.]],
[[24., 34., 44.],
[25., 35., 45.]]]]]), size=(2, 3, 4, 9), nnz=4,
dtype=torch.float64, layout=torch.sparse_bsr)
# _crow_indices
tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]], dtype=torch.int32)
# _col_indices
tensor([[[0, 1, 0, 1],
[0, 1, 2, 0],
[0, 0, 1, 2]],
[[1, 0, 1, 2],
[0, 2, 0, 1],
[0, 1, 2, 1]]], dtype=torch.int32)
# _values
tensor([[[[[ 1., 11., 21.],
[ 2., 12., 22.]],
[[ 2., 12., 22.],
[ 3., 13., 23.]],
[[ 3., 13., 23.],
[ 4., 14., 24.]],
[[ 4., 14., 24.],
[ 5., 15., 25.]]],
[[[ 5., 15., 25.],
[ 6., 16., 26.]],
[[ 6., 16., 26.],
[ 7., 17., 27.]],
[[ 7., 17., 27.],
[ 8., 18., 28.]],
[[ 8., 18., 28.],
[ 9., 19., 29.]]],
[[[ 9., 19., 29.],
[10., 20., 30.]],
[[10., 20., 30.],
[11., 21., 31.]],
[[11., 21., 31.],
[12., 22., 32.]],
[[12., 22., 32.],
[13., 23., 33.]]]],
[[[[13., 23., 33.],
[14., 24., 34.]],
[[14., 24., 34.],
[15., 25., 35.]],
[[15., 25., 35.],
[16., 26., 36.]],
[[16., 26., 36.],
[17., 27., 37.]]],
[[[17., 27., 37.],
[18., 28., 38.]],
[[18., 28., 38.],
[19., 29., 39.]],
[[19., 29., 39.],
[20., 30., 40.]],
[[20., 30., 40.],
[21., 31., 41.]]],
[[[21., 31., 41.],
[22., 32., 42.]],
[[22., 32., 42.],
[23., 33., 43.]],
[[23., 33., 43.],
[24., 34., 44.]],
[[24., 34., 44.],
[25., 35., 45.]]]]], dtype=torch.float64)
########## torch.float32/torch.int64/size=()+(4, 3)+() ##########
# sparse tensor
tensor(crow_indices=tensor([0, 2, 4]),
col_indices=tensor([0, 1, 0, 2]),
values=tensor([[[1.],
[2.]],
[[2.],
[3.]],
[[3.],
[4.]],
[[4.],
[5.]]]), size=(4, 3), nnz=4, layout=torch.sparse_bsr)
# _crow_indices
tensor([0, 2, 4])
# _col_indices
tensor([0, 1, 0, 2])
# _values
tensor([[[1.],
[2.]],
[[2.],
[3.]],
[[3.],
[4.]],
[[4.],
[5.]]])
########## torch.float32/torch.int64/size=()+(0, 0)+() ##########
# sparse tensor
tensor(crow_indices=tensor([0]),
col_indices=tensor([], size=(0,)),
values=tensor([], size=(0, 2, 1)), size=(0, 0), nnz=0,
layout=torch.sparse_bsr)
# _crow_indices
tensor([0])
# _col_indices
tensor([], dtype=torch.int64)
# _values
tensor([], size=(0, 2, 1))
########## torch.float32/torch.int64/size=(2,)+(2, 6)+() ##########
# sparse tensor
tensor(crow_indices=tensor([[0, 2, 4],
[0, 3, 4]]),
col_indices=tensor([[0, 1, 0, 1],
[0, 1, 2, 0]]),
values=tensor([[[[ 1., 11.]],
[[ 2., 12.]],
[[ 3., 13.]],
[[ 4., 14.]]],
[[[ 5., 15.]],
[[ 6., 16.]],
[[ 7., 17.]],
[[ 8., 18.]]]]), size=(2, 2, 6), nnz=4,
layout=torch.sparse_bsr)
# _crow_indices
tensor([[0, 2, 4],
[0, 3, 4]])
# _col_indices
tensor([[0, 1, 0, 1],
[0, 1, 2, 0]])
# _values
tensor([[[[ 1., 11.]],
[[ 2., 12.]],
[[ 3., 13.]],
[[ 4., 14.]]],
[[[ 5., 15.]],
[[ 6., 16.]],
[[ 7., 17.]],
[[ 8., 18.]]]])
########## torch.float32/torch.int64/size=(2, 3)+(4, 9)+() ##########
# sparse tensor
tensor(crow_indices=tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]]),
col_indices=tensor([[[0, 1, 0, 1],
[0, 1, 2, 0],
[0, 0, 1, 2]],
[[1, 0, 1, 2],
[0, 2, 0, 1],
[0, 1, 2, 1]]]),
values=tensor([[[[[ 1., 11., 21.],
[ 2., 12., 22.]],
[[ 2., 12., 22.],
[ 3., 13., 23.]],
[[ 3., 13., 23.],
[ 4., 14., 24.]],
[[ 4., 14., 24.],
[ 5., 15., 25.]]],
[[[ 5., 15., 25.],
[ 6., 16., 26.]],
[[ 6., 16., 26.],
[ 7., 17., 27.]],
[[ 7., 17., 27.],
[ 8., 18., 28.]],
[[ 8., 18., 28.],
[ 9., 19., 29.]]],
[[[ 9., 19., 29.],
[10., 20., 30.]],
[[10., 20., 30.],
[11., 21., 31.]],
[[11., 21., 31.],
[12., 22., 32.]],
[[12., 22., 32.],
[13., 23., 33.]]]],
[[[[13., 23., 33.],
[14., 24., 34.]],
[[14., 24., 34.],
[15., 25., 35.]],
[[15., 25., 35.],
[16., 26., 36.]],
[[16., 26., 36.],
[17., 27., 37.]]],
[[[17., 27., 37.],
[18., 28., 38.]],
[[18., 28., 38.],
[19., 29., 39.]],
[[19., 29., 39.],
[20., 30., 40.]],
[[20., 30., 40.],
[21., 31., 41.]]],
[[[21., 31., 41.],
[22., 32., 42.]],
[[22., 32., 42.],
[23., 33., 43.]],
[[23., 33., 43.],
[24., 34., 44.]],
[[24., 34., 44.],
[25., 35., 45.]]]]]), size=(2, 3, 4, 9), nnz=4,
layout=torch.sparse_bsr)
# _crow_indices
tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]])
# _col_indices
tensor([[[0, 1, 0, 1],
[0, 1, 2, 0],
[0, 0, 1, 2]],
[[1, 0, 1, 2],
[0, 2, 0, 1],
[0, 1, 2, 1]]])
# _values
tensor([[[[[ 1., 11., 21.],
[ 2., 12., 22.]],
[[ 2., 12., 22.],
[ 3., 13., 23.]],
[[ 3., 13., 23.],
[ 4., 14., 24.]],
[[ 4., 14., 24.],
[ 5., 15., 25.]]],
[[[ 5., 15., 25.],
[ 6., 16., 26.]],
[[ 6., 16., 26.],
[ 7., 17., 27.]],
[[ 7., 17., 27.],
[ 8., 18., 28.]],
[[ 8., 18., 28.],
[ 9., 19., 29.]]],
[[[ 9., 19., 29.],
[10., 20., 30.]],
[[10., 20., 30.],
[11., 21., 31.]],
[[11., 21., 31.],
[12., 22., 32.]],
[[12., 22., 32.],
[13., 23., 33.]]]],
[[[[13., 23., 33.],
[14., 24., 34.]],
[[14., 24., 34.],
[15., 25., 35.]],
[[15., 25., 35.],
[16., 26., 36.]],
[[16., 26., 36.],
[17., 27., 37.]]],
[[[17., 27., 37.],
[18., 28., 38.]],
[[18., 28., 38.],
[19., 29., 39.]],
[[19., 29., 39.],
[20., 30., 40.]],
[[20., 30., 40.],
[21., 31., 41.]]],
[[[21., 31., 41.],
[22., 32., 42.]],
[[22., 32., 42.],
[23., 33., 43.]],
[[23., 33., 43.],
[24., 34., 44.]],
[[24., 34., 44.],
[25., 35., 45.]]]]])
########## torch.float64/torch.int64/size=()+(4, 3)+() ##########
# sparse tensor
tensor(crow_indices=tensor([0, 2, 4]),
col_indices=tensor([0, 1, 0, 2]),
values=tensor([[[1.],
[2.]],
[[2.],
[3.]],
[[3.],
[4.]],
[[4.],
[5.]]]), size=(4, 3), nnz=4, dtype=torch.float64,
layout=torch.sparse_bsr)
# _crow_indices
tensor([0, 2, 4])
# _col_indices
tensor([0, 1, 0, 2])
# _values
tensor([[[1.],
[2.]],
[[2.],
[3.]],
[[3.],
[4.]],
[[4.],
[5.]]], dtype=torch.float64)
########## torch.float64/torch.int64/size=()+(0, 0)+() ##########
# sparse tensor
tensor(crow_indices=tensor([0]),
col_indices=tensor([], size=(0,)),
values=tensor([], size=(0, 2, 1)), size=(0, 0), nnz=0,
dtype=torch.float64, layout=torch.sparse_bsr)
# _crow_indices
tensor([0])
# _col_indices
tensor([], dtype=torch.int64)
# _values
tensor([], size=(0, 2, 1), dtype=torch.float64)
########## torch.float64/torch.int64/size=(2,)+(2, 6)+() ##########
# sparse tensor
tensor(crow_indices=tensor([[0, 2, 4],
[0, 3, 4]]),
col_indices=tensor([[0, 1, 0, 1],
[0, 1, 2, 0]]),
values=tensor([[[[ 1., 11.]],
[[ 2., 12.]],
[[ 3., 13.]],
[[ 4., 14.]]],
[[[ 5., 15.]],
[[ 6., 16.]],
[[ 7., 17.]],
[[ 8., 18.]]]]), size=(2, 2, 6), nnz=4,
dtype=torch.float64, layout=torch.sparse_bsr)
# _crow_indices
tensor([[0, 2, 4],
[0, 3, 4]])
# _col_indices
tensor([[0, 1, 0, 1],
[0, 1, 2, 0]])
# _values
tensor([[[[ 1., 11.]],
[[ 2., 12.]],
[[ 3., 13.]],
[[ 4., 14.]]],
[[[ 5., 15.]],
[[ 6., 16.]],
[[ 7., 17.]],
[[ 8., 18.]]]], dtype=torch.float64)
########## torch.float64/torch.int64/size=(2, 3)+(4, 9)+() ##########
# sparse tensor
tensor(crow_indices=tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]]),
col_indices=tensor([[[0, 1, 0, 1],
[0, 1, 2, 0],
[0, 0, 1, 2]],
[[1, 0, 1, 2],
[0, 2, 0, 1],
[0, 1, 2, 1]]]),
values=tensor([[[[[ 1., 11., 21.],
[ 2., 12., 22.]],
[[ 2., 12., 22.],
[ 3., 13., 23.]],
[[ 3., 13., 23.],
[ 4., 14., 24.]],
[[ 4., 14., 24.],
[ 5., 15., 25.]]],
[[[ 5., 15., 25.],
[ 6., 16., 26.]],
[[ 6., 16., 26.],
[ 7., 17., 27.]],
[[ 7., 17., 27.],
[ 8., 18., 28.]],
[[ 8., 18., 28.],
[ 9., 19., 29.]]],
[[[ 9., 19., 29.],
[10., 20., 30.]],
[[10., 20., 30.],
[11., 21., 31.]],
[[11., 21., 31.],
[12., 22., 32.]],
[[12., 22., 32.],
[13., 23., 33.]]]],
[[[[13., 23., 33.],
[14., 24., 34.]],
[[14., 24., 34.],
[15., 25., 35.]],
[[15., 25., 35.],
[16., 26., 36.]],
[[16., 26., 36.],
[17., 27., 37.]]],
[[[17., 27., 37.],
[18., 28., 38.]],
[[18., 28., 38.],
[19., 29., 39.]],
[[19., 29., 39.],
[20., 30., 40.]],
[[20., 30., 40.],
[21., 31., 41.]]],
[[[21., 31., 41.],
[22., 32., 42.]],
[[22., 32., 42.],
[23., 33., 43.]],
[[23., 33., 43.],
[24., 34., 44.]],
[[24., 34., 44.],
[25., 35., 45.]]]]]), size=(2, 3, 4, 9), nnz=4,
dtype=torch.float64, layout=torch.sparse_bsr)
# _crow_indices
tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]])
# _col_indices
tensor([[[0, 1, 0, 1],
[0, 1, 2, 0],
[0, 0, 1, 2]],
[[1, 0, 1, 2],
[0, 2, 0, 1],
[0, 1, 2, 1]]])
# _values
tensor([[[[[ 1., 11., 21.],
[ 2., 12., 22.]],
[[ 2., 12., 22.],
[ 3., 13., 23.]],
[[ 3., 13., 23.],
[ 4., 14., 24.]],
[[ 4., 14., 24.],
[ 5., 15., 25.]]],
[[[ 5., 15., 25.],
[ 6., 16., 26.]],
[[ 6., 16., 26.],
[ 7., 17., 27.]],
[[ 7., 17., 27.],
[ 8., 18., 28.]],
[[ 8., 18., 28.],
[ 9., 19., 29.]]],
[[[ 9., 19., 29.],
[10., 20., 30.]],
[[10., 20., 30.],
[11., 21., 31.]],
[[11., 21., 31.],
[12., 22., 32.]],
[[12., 22., 32.],
[13., 23., 33.]]]],
[[[[13., 23., 33.],
[14., 24., 34.]],
[[14., 24., 34.],
[15., 25., 35.]],
[[15., 25., 35.],
[16., 26., 36.]],
[[16., 26., 36.],
[17., 27., 37.]]],
[[[17., 27., 37.],
[18., 28., 38.]],
[[18., 28., 38.],
[19., 29., 39.]],
[[19., 29., 39.],
[20., 30., 40.]],
[[20., 30., 40.],
[21., 31., 41.]]],
[[[21., 31., 41.],
[22., 32., 42.]],
[[22., 32., 42.],
[23., 33., 43.]],
[[23., 33., 43.],
[24., 34., 44.]],
[[24., 34., 44.],
[25., 35., 45.]]]]], dtype=torch.float64)
########## torch.float32/torch.int32/size=()+(6, 6)+(2,) ##########
# sparse tensor
tensor(crow_indices=tensor([0, 2, 4]),
col_indices=tensor([0, 1, 0, 2]),
values=tensor([[[[ 1., 101.],
[ 11., 111.]],
[[ 2., 102.],
[ 12., 112.]],
[[ 3., 103.],
[ 13., 113.]]],
[[[ 2., 102.],
[ 12., 112.]],
[[ 3., 103.],
[ 13., 113.]],
[[ 4., 104.],
[ 14., 114.]]],
[[[ 3., 103.],
[ 13., 113.]],
[[ 4., 104.],
[ 14., 114.]],
[[ 5., 105.],
[ 15., 115.]]],
[[[ 4., 104.],
[ 14., 114.]],
[[ 5., 105.],
[ 15., 115.]],
[[ 6., 106.],
[ 16., 116.]]]]), size=(6, 6, 2), nnz=4,
layout=torch.sparse_bsr)
# _crow_indices
tensor([0, 2, 4], dtype=torch.int32)
# _col_indices
tensor([0, 1, 0, 2], dtype=torch.int32)
# _values
tensor([[[[ 1., 101.],
[ 11., 111.]],
[[ 2., 102.],
[ 12., 112.]],
[[ 3., 103.],
[ 13., 113.]]],
[[[ 2., 102.],
[ 12., 112.]],
[[ 3., 103.],
[ 13., 113.]],
[[ 4., 104.],
[ 14., 114.]]],
[[[ 3., 103.],
[ 13., 113.]],
[[ 4., 104.],
[ 14., 114.]],
[[ 5., 105.],
[ 15., 115.]]],
[[[ 4., 104.],
[ 14., 114.]],
[[ 5., 105.],
[ 15., 115.]],
[[ 6., 106.],
[ 16., 116.]]]])
########## torch.float32/torch.int32/size=()+(4, 9)+(4, 2) ##########
# sparse tensor
tensor(crow_indices=tensor([0, 2, 4]),
col_indices=tensor([0, 1, 0, 2]),
values=tensor([[[[[1.0000e+00, 1.0010e+03],
[1.0100e+02, 1.1010e+03],
[2.0100e+02, 1.2010e+03],
[3.0100e+02, 1.3010e+03]],
[[1.1000e+01, 1.0110e+03],
[1.1100e+02, 1.1110e+03],
[2.1100e+02, 1.2110e+03],
[3.1100e+02, 1.3110e+03]],
[[2.1000e+01, 1.0210e+03],
[1.2100e+02, 1.1210e+03],
[2.2100e+02, 1.2210e+03],
[3.2100e+02, 1.3210e+03]]],
[[[2.0000e+00, 1.0020e+03],
[1.0200e+02, 1.1020e+03],
[2.0200e+02, 1.2020e+03],
[3.0200e+02, 1.3020e+03]],
[[1.2000e+01, 1.0120e+03],
[1.1200e+02, 1.1120e+03],
[2.1200e+02, 1.2120e+03],
[3.1200e+02, 1.3120e+03]],
[[2.2000e+01, 1.0220e+03],
[1.2200e+02, 1.1220e+03],
[2.2200e+02, 1.2220e+03],
[3.2200e+02, 1.3220e+03]]]],
[[[[2.0000e+00, 1.0020e+03],
[1.0200e+02, 1.1020e+03],
[2.0200e+02, 1.2020e+03],
[3.0200e+02, 1.3020e+03]],
[[1.2000e+01, 1.0120e+03],
[1.1200e+02, 1.1120e+03],
[2.1200e+02, 1.2120e+03],
[3.1200e+02, 1.3120e+03]],
[[2.2000e+01, 1.0220e+03],
[1.2200e+02, 1.1220e+03],
[2.2200e+02, 1.2220e+03],
[3.2200e+02, 1.3220e+03]]],
[[[3.0000e+00, 1.0030e+03],
[1.0300e+02, 1.1030e+03],
[2.0300e+02, 1.2030e+03],
[3.0300e+02, 1.3030e+03]],
[[1.3000e+01, 1.0130e+03],
[1.1300e+02, 1.1130e+03],
[2.1300e+02, 1.2130e+03],
[3.1300e+02, 1.3130e+03]],
[[2.3000e+01, 1.0230e+03],
[1.2300e+02, 1.1230e+03],
[2.2300e+02, 1.2230e+03],
[3.2300e+02, 1.3230e+03]]]],
[[[[3.0000e+00, 1.0030e+03],
[1.0300e+02, 1.1030e+03],
[2.0300e+02, 1.2030e+03],
[3.0300e+02, 1.3030e+03]],
[[1.3000e+01, 1.0130e+03],
[1.1300e+02, 1.1130e+03],
[2.1300e+02, 1.2130e+03],
[3.1300e+02, 1.3130e+03]],
[[2.3000e+01, 1.0230e+03],
[1.2300e+02, 1.1230e+03],
[2.2300e+02, 1.2230e+03],
[3.2300e+02, 1.3230e+03]]],
[[[4.0000e+00, 1.0040e+03],
[1.0400e+02, 1.1040e+03],
[2.0400e+02, 1.2040e+03],
[3.0400e+02, 1.3040e+03]],
[[1.4000e+01, 1.0140e+03],
[1.1400e+02, 1.1140e+03],
[2.1400e+02, 1.2140e+03],
[3.1400e+02, 1.3140e+03]],
[[2.4000e+01, 1.0240e+03],
[1.2400e+02, 1.1240e+03],
[2.2400e+02, 1.2240e+03],
[3.2400e+02, 1.3240e+03]]]],
[[[[4.0000e+00, 1.0040e+03],
[1.0400e+02, 1.1040e+03],
[2.0400e+02, 1.2040e+03],
[3.0400e+02, 1.3040e+03]],
[[1.4000e+01, 1.0140e+03],
[1.1400e+02, 1.1140e+03],
[2.1400e+02, 1.2140e+03],
[3.1400e+02, 1.3140e+03]],
[[2.4000e+01, 1.0240e+03],
[1.2400e+02, 1.1240e+03],
[2.2400e+02, 1.2240e+03],
[3.2400e+02, 1.3240e+03]]],
[[[5.0000e+00, 1.0050e+03],
[1.0500e+02, 1.1050e+03],
[2.0500e+02, 1.2050e+03],
[3.0500e+02, 1.3050e+03]],
[[1.5000e+01, 1.0150e+03],
[1.1500e+02, 1.1150e+03],
[2.1500e+02, 1.2150e+03],
[3.1500e+02, 1.3150e+03]],
[[2.5000e+01, 1.0250e+03],
[1.2500e+02, 1.1250e+03],
[2.2500e+02, 1.2250e+03],
[3.2500e+02, 1.3250e+03]]]]]), size=(4, 9, 4, 2),
nnz=4, layout=torch.sparse_bsr)
# _crow_indices
tensor([0, 2, 4], dtype=torch.int32)
# _col_indices
tensor([0, 1, 0, 2], dtype=torch.int32)
# _values
tensor([[[[[1.0000e+00, 1.0010e+03],
[1.0100e+02, 1.1010e+03],
[2.0100e+02, 1.2010e+03],
[3.0100e+02, 1.3010e+03]],
[[1.1000e+01, 1.0110e+03],
[1.1100e+02, 1.1110e+03],
[2.1100e+02, 1.2110e+03],
[3.1100e+02, 1.3110e+03]],
[[2.1000e+01, 1.0210e+03],
[1.2100e+02, 1.1210e+03],
[2.2100e+02, 1.2210e+03],
[3.2100e+02, 1.3210e+03]]],
[[[2.0000e+00, 1.0020e+03],
[1.0200e+02, 1.1020e+03],
[2.0200e+02, 1.2020e+03],
[3.0200e+02, 1.3020e+03]],
[[1.2000e+01, 1.0120e+03],
[1.1200e+02, 1.1120e+03],
[2.1200e+02, 1.2120e+03],
[3.1200e+02, 1.3120e+03]],
[[2.2000e+01, 1.0220e+03],
[1.2200e+02, 1.1220e+03],
[2.2200e+02, 1.2220e+03],
[3.2200e+02, 1.3220e+03]]]],
[[[[2.0000e+00, 1.0020e+03],
[1.0200e+02, 1.1020e+03],
[2.0200e+02, 1.2020e+03],
[3.0200e+02, 1.3020e+03]],
[[1.2000e+01, 1.0120e+03],
[1.1200e+02, 1.1120e+03],
[2.1200e+02, 1.2120e+03],
[3.1200e+02, 1.3120e+03]],
[[2.2000e+01, 1.0220e+03],
[1.2200e+02, 1.1220e+03],
[2.2200e+02, 1.2220e+03],
[3.2200e+02, 1.3220e+03]]],
[[[3.0000e+00, 1.0030e+03],
[1.0300e+02, 1.1030e+03],
[2.0300e+02, 1.2030e+03],
[3.0300e+02, 1.3030e+03]],
[[1.3000e+01, 1.0130e+03],
[1.1300e+02, 1.1130e+03],
[2.1300e+02, 1.2130e+03],
[3.1300e+02, 1.3130e+03]],
[[2.3000e+01, 1.0230e+03],
[1.2300e+02, 1.1230e+03],
[2.2300e+02, 1.2230e+03],
[3.2300e+02, 1.3230e+03]]]],
[[[[3.0000e+00, 1.0030e+03],
[1.0300e+02, 1.1030e+03],
[2.0300e+02, 1.2030e+03],
[3.0300e+02, 1.3030e+03]],
[[1.3000e+01, 1.0130e+03],
[1.1300e+02, 1.1130e+03],
[2.1300e+02, 1.2130e+03],
[3.1300e+02, 1.3130e+03]],
[[2.3000e+01, 1.0230e+03],
[1.2300e+02, 1.1230e+03],
[2.2300e+02, 1.2230e+03],
[3.2300e+02, 1.3230e+03]]],
[[[4.0000e+00, 1.0040e+03],
[1.0400e+02, 1.1040e+03],
[2.0400e+02, 1.2040e+03],
[3.0400e+02, 1.3040e+03]],
[[1.4000e+01, 1.0140e+03],
[1.1400e+02, 1.1140e+03],
[2.1400e+02, 1.2140e+03],
[3.1400e+02, 1.3140e+03]],
[[2.4000e+01, 1.0240e+03],
[1.2400e+02, 1.1240e+03],
[2.2400e+02, 1.2240e+03],
[3.2400e+02, 1.3240e+03]]]],
[[[[4.0000e+00, 1.0040e+03],
[1.0400e+02, 1.1040e+03],
[2.0400e+02, 1.2040e+03],
[3.0400e+02, 1.3040e+03]],
[[1.4000e+01, 1.0140e+03],
[1.1400e+02, 1.1140e+03],
[2.1400e+02, 1.2140e+03],
[3.1400e+02, 1.3140e+03]],
[[2.4000e+01, 1.0240e+03],
[1.2400e+02, 1.1240e+03],
[2.2400e+02, 1.2240e+03],
[3.2400e+02, 1.3240e+03]]],
[[[5.0000e+00, 1.0050e+03],
[1.0500e+02, 1.1050e+03],
[2.0500e+02, 1.2050e+03],
[3.0500e+02, 1.3050e+03]],
[[1.5000e+01, 1.0150e+03],
[1.1500e+02, 1.1150e+03],
[2.1500e+02, 1.2150e+03],
[3.1500e+02, 1.3150e+03]],
[[2.5000e+01, 1.0250e+03],
[1.2500e+02, 1.1250e+03],
[2.2500e+02, 1.2250e+03],
[3.2500e+02, 1.3250e+03]]]]])
########## torch.float32/torch.int32/size=(2, 3)+(6, 6)+(2, 1) ##########
# sparse tensor
tensor(crow_indices=tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]]),
col_indices=tensor([[[0, 1, 0, 1],
[0, 1, 2, 0],
[0, 0, 1, 2]],
[[1, 0, 1, 2],
[0, 2, 0, 1],
[0, 1, 2, 1]]]),
values=tensor([[[[[[[ 1.],
[101.]],
[[ 11.],
[111.]]],
[[[ 2.],
[102.]],
[[ 12.],
[112.]]],
[[[ 3.],
[103.]],
[[ 13.],
[113.]]]],
[[[[ 2.],
[102.]],
[[ 12.],
[112.]]],
[[[ 3.],
[103.]],
[[ 13.],
[113.]]],
[[[ 4.],
[104.]],
[[ 14.],
[114.]]]],
[[[[ 3.],
[103.]],
[[ 13.],
[113.]]],
[[[ 4.],
[104.]],
[[ 14.],
[114.]]],
[[[ 5.],
[105.]],
[[ 15.],
[115.]]]],
[[[[ 4.],
[104.]],
[[ 14.],
[114.]]],
[[[ 5.],
[105.]],
[[ 15.],
[115.]]],
[[[ 6.],
[106.]],
[[ 16.],
[116.]]]]],
[[[[[ 5.],
[105.]],
[[ 15.],
[115.]]],
[[[ 6.],
[106.]],
[[ 16.],
[116.]]],
[[[ 7.],
[107.]],
[[ 17.],
[117.]]]],
[[[[ 6.],
[106.]],
[[ 16.],
[116.]]],
[[[ 7.],
[107.]],
[[ 17.],
[117.]]],
[[[ 8.],
[108.]],
[[ 18.],
[118.]]]],
[[[[ 7.],
[107.]],
[[ 17.],
[117.]]],
[[[ 8.],
[108.]],
[[ 18.],
[118.]]],
[[[ 9.],
[109.]],
[[ 19.],
[119.]]]],
[[[[ 8.],
[108.]],
[[ 18.],
[118.]]],
[[[ 9.],
[109.]],
[[ 19.],
[119.]]],
[[[ 10.],
[110.]],
[[ 20.],
[120.]]]]],
[[[[[ 9.],
[109.]],
[[ 19.],
[119.]]],
[[[ 10.],
[110.]],
[[ 20.],
[120.]]],
[[[ 11.],
[111.]],
[[ 21.],
[121.]]]],
[[[[ 10.],
[110.]],
[[ 20.],
[120.]]],
[[[ 11.],
[111.]],
[[ 21.],
[121.]]],
[[[ 12.],
[112.]],
[[ 22.],
[122.]]]],
[[[[ 11.],
[111.]],
[[ 21.],
[121.]]],
[[[ 12.],
[112.]],
[[ 22.],
[122.]]],
[[[ 13.],
[113.]],
[[ 23.],
[123.]]]],
[[[[ 12.],
[112.]],
[[ 22.],
[122.]]],
[[[ 13.],
[113.]],
[[ 23.],
[123.]]],
[[[ 14.],
[114.]],
[[ 24.],
[124.]]]]]],
[[[[[[ 13.],
[113.]],
[[ 23.],
[123.]]],
[[[ 14.],
[114.]],
[[ 24.],
[124.]]],
[[[ 15.],
[115.]],
[[ 25.],
[125.]]]],
[[[[ 14.],
[114.]],
[[ 24.],
[124.]]],
[[[ 15.],
[115.]],
[[ 25.],
[125.]]],
[[[ 16.],
[116.]],
[[ 26.],
[126.]]]],
[[[[ 15.],
[115.]],
[[ 25.],
[125.]]],
[[[ 16.],
[116.]],
[[ 26.],
[126.]]],
[[[ 17.],
[117.]],
[[ 27.],
[127.]]]],
[[[[ 16.],
[116.]],
[[ 26.],
[126.]]],
[[[ 17.],
[117.]],
[[ 27.],
[127.]]],
[[[ 18.],
[118.]],
[[ 28.],
[128.]]]]],
[[[[[ 17.],
[117.]],
[[ 27.],
[127.]]],
[[[ 18.],
[118.]],
[[ 28.],
[128.]]],
[[[ 19.],
[119.]],
[[ 29.],
[129.]]]],
[[[[ 18.],
[118.]],
[[ 28.],
[128.]]],
[[[ 19.],
[119.]],
[[ 29.],
[129.]]],
[[[ 20.],
[120.]],
[[ 30.],
[130.]]]],
[[[[ 19.],
[119.]],
[[ 29.],
[129.]]],
[[[ 20.],
[120.]],
[[ 30.],
[130.]]],
[[[ 21.],
[121.]],
[[ 31.],
[131.]]]],
[[[[ 20.],
[120.]],
[[ 30.],
[130.]]],
[[[ 21.],
[121.]],
[[ 31.],
[131.]]],
[[[ 22.],
[122.]],
[[ 32.],
[132.]]]]],
[[[[[ 21.],
[121.]],
[[ 31.],
[131.]]],
[[[ 22.],
[122.]],
[[ 32.],
[132.]]],
[[[ 23.],
[123.]],
[[ 33.],
[133.]]]],
[[[[ 22.],
[122.]],
[[ 32.],
[132.]]],
[[[ 23.],
[123.]],
[[ 33.],
[133.]]],
[[[ 24.],
[124.]],
[[ 34.],
[134.]]]],
[[[[ 23.],
[123.]],
[[ 33.],
[133.]]],
[[[ 24.],
[124.]],
[[ 34.],
[134.]]],
[[[ 25.],
[125.]],
[[ 35.],
[135.]]]],
[[[[ 24.],
[124.]],
[[ 34.],
[134.]]],
[[[ 25.],
[125.]],
[[ 35.],
[135.]]],
[[[ 26.],
[126.]],
[[ 36.],
[136.]]]]]]]), size=(2, 3, 6, 6, 2, 1), nnz=4,
layout=torch.sparse_bsr)
# _crow_indices
tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]], dtype=torch.int32)
# _col_indices
tensor([[[0, 1, 0, 1],
[0, 1, 2, 0],
[0, 0, 1, 2]],
[[1, 0, 1, 2],
[0, 2, 0, 1],
[0, 1, 2, 1]]], dtype=torch.int32)
# _values
tensor([[[[[[[ 1.],
[101.]],
[[ 11.],
[111.]]],
[[[ 2.],
[102.]],
[[ 12.],
[112.]]],
[[[ 3.],
[103.]],
[[ 13.],
[113.]]]],
[[[[ 2.],
[102.]],
[[ 12.],
[112.]]],
[[[ 3.],
[103.]],
[[ 13.],
[113.]]],
[[[ 4.],
[104.]],
[[ 14.],
[114.]]]],
[[[[ 3.],
[103.]],
[[ 13.],
[113.]]],
[[[ 4.],
[104.]],
[[ 14.],
[114.]]],
[[[ 5.],
[105.]],
[[ 15.],
[115.]]]],
[[[[ 4.],
[104.]],
[[ 14.],
[114.]]],
[[[ 5.],
[105.]],
[[ 15.],
[115.]]],
[[[ 6.],
[106.]],
[[ 16.],
[116.]]]]],
[[[[[ 5.],
[105.]],
[[ 15.],
[115.]]],
[[[ 6.],
[106.]],
[[ 16.],
[116.]]],
[[[ 7.],
[107.]],
[[ 17.],
[117.]]]],
[[[[ 6.],
[106.]],
[[ 16.],
[116.]]],
[[[ 7.],
[107.]],
[[ 17.],
[117.]]],
[[[ 8.],
[108.]],
[[ 18.],
[118.]]]],
[[[[ 7.],
[107.]],
[[ 17.],
[117.]]],
[[[ 8.],
[108.]],
[[ 18.],
[118.]]],
[[[ 9.],
[109.]],
[[ 19.],
[119.]]]],
[[[[ 8.],
[108.]],
[[ 18.],
[118.]]],
[[[ 9.],
[109.]],
[[ 19.],
[119.]]],
[[[ 10.],
[110.]],
[[ 20.],
[120.]]]]],
[[[[[ 9.],
[109.]],
[[ 19.],
[119.]]],
[[[ 10.],
[110.]],
[[ 20.],
[120.]]],
[[[ 11.],
[111.]],
[[ 21.],
[121.]]]],
[[[[ 10.],
[110.]],
[[ 20.],
[120.]]],
[[[ 11.],
[111.]],
[[ 21.],
[121.]]],
[[[ 12.],
[112.]],
[[ 22.],
[122.]]]],
[[[[ 11.],
[111.]],
[[ 21.],
[121.]]],
[[[ 12.],
[112.]],
[[ 22.],
[122.]]],
[[[ 13.],
[113.]],
[[ 23.],
[123.]]]],
[[[[ 12.],
[112.]],
[[ 22.],
[122.]]],
[[[ 13.],
[113.]],
[[ 23.],
[123.]]],
[[[ 14.],
[114.]],
[[ 24.],
[124.]]]]]],
[[[[[[ 13.],
[113.]],
[[ 23.],
[123.]]],
[[[ 14.],
[114.]],
[[ 24.],
[124.]]],
[[[ 15.],
[115.]],
[[ 25.],
[125.]]]],
[[[[ 14.],
[114.]],
[[ 24.],
[124.]]],
[[[ 15.],
[115.]],
[[ 25.],
[125.]]],
[[[ 16.],
[116.]],
[[ 26.],
[126.]]]],
[[[[ 15.],
[115.]],
[[ 25.],
[125.]]],
[[[ 16.],
[116.]],
[[ 26.],
[126.]]],
[[[ 17.],
[117.]],
[[ 27.],
[127.]]]],
[[[[ 16.],
[116.]],
[[ 26.],
[126.]]],
[[[ 17.],
[117.]],
[[ 27.],
[127.]]],
[[[ 18.],
[118.]],
[[ 28.],
[128.]]]]],
[[[[[ 17.],
[117.]],
[[ 27.],
[127.]]],
[[[ 18.],
[118.]],
[[ 28.],
[128.]]],
[[[ 19.],
[119.]],
[[ 29.],
[129.]]]],
[[[[ 18.],
[118.]],
[[ 28.],
[128.]]],
[[[ 19.],
[119.]],
[[ 29.],
[129.]]],
[[[ 20.],
[120.]],
[[ 30.],
[130.]]]],
[[[[ 19.],
[119.]],
[[ 29.],
[129.]]],
[[[ 20.],
[120.]],
[[ 30.],
[130.]]],
[[[ 21.],
[121.]],
[[ 31.],
[131.]]]],
[[[[ 20.],
[120.]],
[[ 30.],
[130.]]],
[[[ 21.],
[121.]],
[[ 31.],
[131.]]],
[[[ 22.],
[122.]],
[[ 32.],
[132.]]]]],
[[[[[ 21.],
[121.]],
[[ 31.],
[131.]]],
[[[ 22.],
[122.]],
[[ 32.],
[132.]]],
[[[ 23.],
[123.]],
[[ 33.],
[133.]]]],
[[[[ 22.],
[122.]],
[[ 32.],
[132.]]],
[[[ 23.],
[123.]],
[[ 33.],
[133.]]],
[[[ 24.],
[124.]],
[[ 34.],
[134.]]]],
[[[[ 23.],
[123.]],
[[ 33.],
[133.]]],
[[[ 24.],
[124.]],
[[ 34.],
[134.]]],
[[[ 25.],
[125.]],
[[ 35.],
[135.]]]],
[[[[ 24.],
[124.]],
[[ 34.],
[134.]]],
[[[ 25.],
[125.]],
[[ 35.],
[135.]]],
[[[ 26.],
[126.]],
[[ 36.],
[136.]]]]]]])
########## torch.float64/torch.int32/size=()+(6, 6)+(2,) ##########
# sparse tensor
tensor(crow_indices=tensor([0, 2, 4]),
col_indices=tensor([0, 1, 0, 2]),
values=tensor([[[[ 1., 101.],
[ 11., 111.]],
[[ 2., 102.],
[ 12., 112.]],
[[ 3., 103.],
[ 13., 113.]]],
[[[ 2., 102.],
[ 12., 112.]],
[[ 3., 103.],
[ 13., 113.]],
[[ 4., 104.],
[ 14., 114.]]],
[[[ 3., 103.],
[ 13., 113.]],
[[ 4., 104.],
[ 14., 114.]],
[[ 5., 105.],
[ 15., 115.]]],
[[[ 4., 104.],
[ 14., 114.]],
[[ 5., 105.],
[ 15., 115.]],
[[ 6., 106.],
[ 16., 116.]]]]), size=(6, 6, 2), nnz=4,
dtype=torch.float64, layout=torch.sparse_bsr)
# _crow_indices
tensor([0, 2, 4], dtype=torch.int32)
# _col_indices
tensor([0, 1, 0, 2], dtype=torch.int32)
# _values
tensor([[[[ 1., 101.],
[ 11., 111.]],
[[ 2., 102.],
[ 12., 112.]],
[[ 3., 103.],
[ 13., 113.]]],
[[[ 2., 102.],
[ 12., 112.]],
[[ 3., 103.],
[ 13., 113.]],
[[ 4., 104.],
[ 14., 114.]]],
[[[ 3., 103.],
[ 13., 113.]],
[[ 4., 104.],
[ 14., 114.]],
[[ 5., 105.],
[ 15., 115.]]],
[[[ 4., 104.],
[ 14., 114.]],
[[ 5., 105.],
[ 15., 115.]],
[[ 6., 106.],
[ 16., 116.]]]], dtype=torch.float64)
########## torch.float64/torch.int32/size=()+(4, 9)+(4, 2) ##########
# sparse tensor
tensor(crow_indices=tensor([0, 2, 4]),
col_indices=tensor([0, 1, 0, 2]),
values=tensor([[[[[1.0000e+00, 1.0010e+03],
[1.0100e+02, 1.1010e+03],
[2.0100e+02, 1.2010e+03],
[3.0100e+02, 1.3010e+03]],
[[1.1000e+01, 1.0110e+03],
[1.1100e+02, 1.1110e+03],
[2.1100e+02, 1.2110e+03],
[3.1100e+02, 1.3110e+03]],
[[2.1000e+01, 1.0210e+03],
[1.2100e+02, 1.1210e+03],
[2.2100e+02, 1.2210e+03],
[3.2100e+02, 1.3210e+03]]],
[[[2.0000e+00, 1.0020e+03],
[1.0200e+02, 1.1020e+03],
[2.0200e+02, 1.2020e+03],
[3.0200e+02, 1.3020e+03]],
[[1.2000e+01, 1.0120e+03],
[1.1200e+02, 1.1120e+03],
[2.1200e+02, 1.2120e+03],
[3.1200e+02, 1.3120e+03]],
[[2.2000e+01, 1.0220e+03],
[1.2200e+02, 1.1220e+03],
[2.2200e+02, 1.2220e+03],
[3.2200e+02, 1.3220e+03]]]],
[[[[2.0000e+00, 1.0020e+03],
[1.0200e+02, 1.1020e+03],
[2.0200e+02, 1.2020e+03],
[3.0200e+02, 1.3020e+03]],
[[1.2000e+01, 1.0120e+03],
[1.1200e+02, 1.1120e+03],
[2.1200e+02, 1.2120e+03],
[3.1200e+02, 1.3120e+03]],
[[2.2000e+01, 1.0220e+03],
[1.2200e+02, 1.1220e+03],
[2.2200e+02, 1.2220e+03],
[3.2200e+02, 1.3220e+03]]],
[[[3.0000e+00, 1.0030e+03],
[1.0300e+02, 1.1030e+03],
[2.0300e+02, 1.2030e+03],
[3.0300e+02, 1.3030e+03]],
[[1.3000e+01, 1.0130e+03],
[1.1300e+02, 1.1130e+03],
[2.1300e+02, 1.2130e+03],
[3.1300e+02, 1.3130e+03]],
[[2.3000e+01, 1.0230e+03],
[1.2300e+02, 1.1230e+03],
[2.2300e+02, 1.2230e+03],
[3.2300e+02, 1.3230e+03]]]],
[[[[3.0000e+00, 1.0030e+03],
[1.0300e+02, 1.1030e+03],
[2.0300e+02, 1.2030e+03],
[3.0300e+02, 1.3030e+03]],
[[1.3000e+01, 1.0130e+03],
[1.1300e+02, 1.1130e+03],
[2.1300e+02, 1.2130e+03],
[3.1300e+02, 1.3130e+03]],
[[2.3000e+01, 1.0230e+03],
[1.2300e+02, 1.1230e+03],
[2.2300e+02, 1.2230e+03],
[3.2300e+02, 1.3230e+03]]],
[[[4.0000e+00, 1.0040e+03],
[1.0400e+02, 1.1040e+03],
[2.0400e+02, 1.2040e+03],
[3.0400e+02, 1.3040e+03]],
[[1.4000e+01, 1.0140e+03],
[1.1400e+02, 1.1140e+03],
[2.1400e+02, 1.2140e+03],
[3.1400e+02, 1.3140e+03]],
[[2.4000e+01, 1.0240e+03],
[1.2400e+02, 1.1240e+03],
[2.2400e+02, 1.2240e+03],
[3.2400e+02, 1.3240e+03]]]],
[[[[4.0000e+00, 1.0040e+03],
[1.0400e+02, 1.1040e+03],
[2.0400e+02, 1.2040e+03],
[3.0400e+02, 1.3040e+03]],
[[1.4000e+01, 1.0140e+03],
[1.1400e+02, 1.1140e+03],
[2.1400e+02, 1.2140e+03],
[3.1400e+02, 1.3140e+03]],
[[2.4000e+01, 1.0240e+03],
[1.2400e+02, 1.1240e+03],
[2.2400e+02, 1.2240e+03],
[3.2400e+02, 1.3240e+03]]],
[[[5.0000e+00, 1.0050e+03],
[1.0500e+02, 1.1050e+03],
[2.0500e+02, 1.2050e+03],
[3.0500e+02, 1.3050e+03]],
[[1.5000e+01, 1.0150e+03],
[1.1500e+02, 1.1150e+03],
[2.1500e+02, 1.2150e+03],
[3.1500e+02, 1.3150e+03]],
[[2.5000e+01, 1.0250e+03],
[1.2500e+02, 1.1250e+03],
[2.2500e+02, 1.2250e+03],
[3.2500e+02, 1.3250e+03]]]]]), size=(4, 9, 4, 2),
nnz=4, dtype=torch.float64, layout=torch.sparse_bsr)
# _crow_indices
tensor([0, 2, 4], dtype=torch.int32)
# _col_indices
tensor([0, 1, 0, 2], dtype=torch.int32)
# _values
tensor([[[[[1.0000e+00, 1.0010e+03],
[1.0100e+02, 1.1010e+03],
[2.0100e+02, 1.2010e+03],
[3.0100e+02, 1.3010e+03]],
[[1.1000e+01, 1.0110e+03],
[1.1100e+02, 1.1110e+03],
[2.1100e+02, 1.2110e+03],
[3.1100e+02, 1.3110e+03]],
[[2.1000e+01, 1.0210e+03],
[1.2100e+02, 1.1210e+03],
[2.2100e+02, 1.2210e+03],
[3.2100e+02, 1.3210e+03]]],
[[[2.0000e+00, 1.0020e+03],
[1.0200e+02, 1.1020e+03],
[2.0200e+02, 1.2020e+03],
[3.0200e+02, 1.3020e+03]],
[[1.2000e+01, 1.0120e+03],
[1.1200e+02, 1.1120e+03],
[2.1200e+02, 1.2120e+03],
[3.1200e+02, 1.3120e+03]],
[[2.2000e+01, 1.0220e+03],
[1.2200e+02, 1.1220e+03],
[2.2200e+02, 1.2220e+03],
[3.2200e+02, 1.3220e+03]]]],
[[[[2.0000e+00, 1.0020e+03],
[1.0200e+02, 1.1020e+03],
[2.0200e+02, 1.2020e+03],
[3.0200e+02, 1.3020e+03]],
[[1.2000e+01, 1.0120e+03],
[1.1200e+02, 1.1120e+03],
[2.1200e+02, 1.2120e+03],
[3.1200e+02, 1.3120e+03]],
[[2.2000e+01, 1.0220e+03],
[1.2200e+02, 1.1220e+03],
[2.2200e+02, 1.2220e+03],
[3.2200e+02, 1.3220e+03]]],
[[[3.0000e+00, 1.0030e+03],
[1.0300e+02, 1.1030e+03],
[2.0300e+02, 1.2030e+03],
[3.0300e+02, 1.3030e+03]],
[[1.3000e+01, 1.0130e+03],
[1.1300e+02, 1.1130e+03],
[2.1300e+02, 1.2130e+03],
[3.1300e+02, 1.3130e+03]],
[[2.3000e+01, 1.0230e+03],
[1.2300e+02, 1.1230e+03],
[2.2300e+02, 1.2230e+03],
[3.2300e+02, 1.3230e+03]]]],
[[[[3.0000e+00, 1.0030e+03],
[1.0300e+02, 1.1030e+03],
[2.0300e+02, 1.2030e+03],
[3.0300e+02, 1.3030e+03]],
[[1.3000e+01, 1.0130e+03],
[1.1300e+02, 1.1130e+03],
[2.1300e+02, 1.2130e+03],
[3.1300e+02, 1.3130e+03]],
[[2.3000e+01, 1.0230e+03],
[1.2300e+02, 1.1230e+03],
[2.2300e+02, 1.2230e+03],
[3.2300e+02, 1.3230e+03]]],
[[[4.0000e+00, 1.0040e+03],
[1.0400e+02, 1.1040e+03],
[2.0400e+02, 1.2040e+03],
[3.0400e+02, 1.3040e+03]],
[[1.4000e+01, 1.0140e+03],
[1.1400e+02, 1.1140e+03],
[2.1400e+02, 1.2140e+03],
[3.1400e+02, 1.3140e+03]],
[[2.4000e+01, 1.0240e+03],
[1.2400e+02, 1.1240e+03],
[2.2400e+02, 1.2240e+03],
[3.2400e+02, 1.3240e+03]]]],
[[[[4.0000e+00, 1.0040e+03],
[1.0400e+02, 1.1040e+03],
[2.0400e+02, 1.2040e+03],
[3.0400e+02, 1.3040e+03]],
[[1.4000e+01, 1.0140e+03],
[1.1400e+02, 1.1140e+03],
[2.1400e+02, 1.2140e+03],
[3.1400e+02, 1.3140e+03]],
[[2.4000e+01, 1.0240e+03],
[1.2400e+02, 1.1240e+03],
[2.2400e+02, 1.2240e+03],
[3.2400e+02, 1.3240e+03]]],
[[[5.0000e+00, 1.0050e+03],
[1.0500e+02, 1.1050e+03],
[2.0500e+02, 1.2050e+03],
[3.0500e+02, 1.3050e+03]],
[[1.5000e+01, 1.0150e+03],
[1.1500e+02, 1.1150e+03],
[2.1500e+02, 1.2150e+03],
[3.1500e+02, 1.3150e+03]],
[[2.5000e+01, 1.0250e+03],
[1.2500e+02, 1.1250e+03],
[2.2500e+02, 1.2250e+03],
[3.2500e+02, 1.3250e+03]]]]], dtype=torch.float64)
########## torch.float64/torch.int32/size=(2, 3)+(6, 6)+(2, 1) ##########
# sparse tensor
tensor(crow_indices=tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]]),
col_indices=tensor([[[0, 1, 0, 1],
[0, 1, 2, 0],
[0, 0, 1, 2]],
[[1, 0, 1, 2],
[0, 2, 0, 1],
[0, 1, 2, 1]]]),
values=tensor([[[[[[[ 1.],
[101.]],
[[ 11.],
[111.]]],
[[[ 2.],
[102.]],
[[ 12.],
[112.]]],
[[[ 3.],
[103.]],
[[ 13.],
[113.]]]],
[[[[ 2.],
[102.]],
[[ 12.],
[112.]]],
[[[ 3.],
[103.]],
[[ 13.],
[113.]]],
[[[ 4.],
[104.]],
[[ 14.],
[114.]]]],
[[[[ 3.],
[103.]],
[[ 13.],
[113.]]],
[[[ 4.],
[104.]],
[[ 14.],
[114.]]],
[[[ 5.],
[105.]],
[[ 15.],
[115.]]]],
[[[[ 4.],
[104.]],
[[ 14.],
[114.]]],
[[[ 5.],
[105.]],
[[ 15.],
[115.]]],
[[[ 6.],
[106.]],
[[ 16.],
[116.]]]]],
[[[[[ 5.],
[105.]],
[[ 15.],
[115.]]],
[[[ 6.],
[106.]],
[[ 16.],
[116.]]],
[[[ 7.],
[107.]],
[[ 17.],
[117.]]]],
[[[[ 6.],
[106.]],
[[ 16.],
[116.]]],
[[[ 7.],
[107.]],
[[ 17.],
[117.]]],
[[[ 8.],
[108.]],
[[ 18.],
[118.]]]],
[[[[ 7.],
[107.]],
[[ 17.],
[117.]]],
[[[ 8.],
[108.]],
[[ 18.],
[118.]]],
[[[ 9.],
[109.]],
[[ 19.],
[119.]]]],
[[[[ 8.],
[108.]],
[[ 18.],
[118.]]],
[[[ 9.],
[109.]],
[[ 19.],
[119.]]],
[[[ 10.],
[110.]],
[[ 20.],
[120.]]]]],
[[[[[ 9.],
[109.]],
[[ 19.],
[119.]]],
[[[ 10.],
[110.]],
[[ 20.],
[120.]]],
[[[ 11.],
[111.]],
[[ 21.],
[121.]]]],
[[[[ 10.],
[110.]],
[[ 20.],
[120.]]],
[[[ 11.],
[111.]],
[[ 21.],
[121.]]],
[[[ 12.],
[112.]],
[[ 22.],
[122.]]]],
[[[[ 11.],
[111.]],
[[ 21.],
[121.]]],
[[[ 12.],
[112.]],
[[ 22.],
[122.]]],
[[[ 13.],
[113.]],
[[ 23.],
[123.]]]],
[[[[ 12.],
[112.]],
[[ 22.],
[122.]]],
[[[ 13.],
[113.]],
[[ 23.],
[123.]]],
[[[ 14.],
[114.]],
[[ 24.],
[124.]]]]]],
[[[[[[ 13.],
[113.]],
[[ 23.],
[123.]]],
[[[ 14.],
[114.]],
[[ 24.],
[124.]]],
[[[ 15.],
[115.]],
[[ 25.],
[125.]]]],
[[[[ 14.],
[114.]],
[[ 24.],
[124.]]],
[[[ 15.],
[115.]],
[[ 25.],
[125.]]],
[[[ 16.],
[116.]],
[[ 26.],
[126.]]]],
[[[[ 15.],
[115.]],
[[ 25.],
[125.]]],
[[[ 16.],
[116.]],
[[ 26.],
[126.]]],
[[[ 17.],
[117.]],
[[ 27.],
[127.]]]],
[[[[ 16.],
[116.]],
[[ 26.],
[126.]]],
[[[ 17.],
[117.]],
[[ 27.],
[127.]]],
[[[ 18.],
[118.]],
[[ 28.],
[128.]]]]],
[[[[[ 17.],
[117.]],
[[ 27.],
[127.]]],
[[[ 18.],
[118.]],
[[ 28.],
[128.]]],
[[[ 19.],
[119.]],
[[ 29.],
[129.]]]],
[[[[ 18.],
[118.]],
[[ 28.],
[128.]]],
[[[ 19.],
[119.]],
[[ 29.],
[129.]]],
[[[ 20.],
[120.]],
[[ 30.],
[130.]]]],
[[[[ 19.],
[119.]],
[[ 29.],
[129.]]],
[[[ 20.],
[120.]],
[[ 30.],
[130.]]],
[[[ 21.],
[121.]],
[[ 31.],
[131.]]]],
[[[[ 20.],
[120.]],
[[ 30.],
[130.]]],
[[[ 21.],
[121.]],
[[ 31.],
[131.]]],
[[[ 22.],
[122.]],
[[ 32.],
[132.]]]]],
[[[[[ 21.],
[121.]],
[[ 31.],
[131.]]],
[[[ 22.],
[122.]],
[[ 32.],
[132.]]],
[[[ 23.],
[123.]],
[[ 33.],
[133.]]]],
[[[[ 22.],
[122.]],
[[ 32.],
[132.]]],
[[[ 23.],
[123.]],
[[ 33.],
[133.]]],
[[[ 24.],
[124.]],
[[ 34.],
[134.]]]],
[[[[ 23.],
[123.]],
[[ 33.],
[133.]]],
[[[ 24.],
[124.]],
[[ 34.],
[134.]]],
[[[ 25.],
[125.]],
[[ 35.],
[135.]]]],
[[[[ 24.],
[124.]],
[[ 34.],
[134.]]],
[[[ 25.],
[125.]],
[[ 35.],
[135.]]],
[[[ 26.],
[126.]],
[[ 36.],
[136.]]]]]]]), size=(2, 3, 6, 6, 2, 1), nnz=4,
dtype=torch.float64, layout=torch.sparse_bsr)
# _crow_indices
tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]], dtype=torch.int32)
# _col_indices
tensor([[[0, 1, 0, 1],
[0, 1, 2, 0],
[0, 0, 1, 2]],
[[1, 0, 1, 2],
[0, 2, 0, 1],
[0, 1, 2, 1]]], dtype=torch.int32)
# _values
tensor([[[[[[[ 1.],
[101.]],
[[ 11.],
[111.]]],
[[[ 2.],
[102.]],
[[ 12.],
[112.]]],
[[[ 3.],
[103.]],
[[ 13.],
[113.]]]],
[[[[ 2.],
[102.]],
[[ 12.],
[112.]]],
[[[ 3.],
[103.]],
[[ 13.],
[113.]]],
[[[ 4.],
[104.]],
[[ 14.],
[114.]]]],
[[[[ 3.],
[103.]],
[[ 13.],
[113.]]],
[[[ 4.],
[104.]],
[[ 14.],
[114.]]],
[[[ 5.],
[105.]],
[[ 15.],
[115.]]]],
[[[[ 4.],
[104.]],
[[ 14.],
[114.]]],
[[[ 5.],
[105.]],
[[ 15.],
[115.]]],
[[[ 6.],
[106.]],
[[ 16.],
[116.]]]]],
[[[[[ 5.],
[105.]],
[[ 15.],
[115.]]],
[[[ 6.],
[106.]],
[[ 16.],
[116.]]],
[[[ 7.],
[107.]],
[[ 17.],
[117.]]]],
[[[[ 6.],
[106.]],
[[ 16.],
[116.]]],
[[[ 7.],
[107.]],
[[ 17.],
[117.]]],
[[[ 8.],
[108.]],
[[ 18.],
[118.]]]],
[[[[ 7.],
[107.]],
[[ 17.],
[117.]]],
[[[ 8.],
[108.]],
[[ 18.],
[118.]]],
[[[ 9.],
[109.]],
[[ 19.],
[119.]]]],
[[[[ 8.],
[108.]],
[[ 18.],
[118.]]],
[[[ 9.],
[109.]],
[[ 19.],
[119.]]],
[[[ 10.],
[110.]],
[[ 20.],
[120.]]]]],
[[[[[ 9.],
[109.]],
[[ 19.],
[119.]]],
[[[ 10.],
[110.]],
[[ 20.],
[120.]]],
[[[ 11.],
[111.]],
[[ 21.],
[121.]]]],
[[[[ 10.],
[110.]],
[[ 20.],
[120.]]],
[[[ 11.],
[111.]],
[[ 21.],
[121.]]],
[[[ 12.],
[112.]],
[[ 22.],
[122.]]]],
[[[[ 11.],
[111.]],
[[ 21.],
[121.]]],
[[[ 12.],
[112.]],
[[ 22.],
[122.]]],
[[[ 13.],
[113.]],
[[ 23.],
[123.]]]],
[[[[ 12.],
[112.]],
[[ 22.],
[122.]]],
[[[ 13.],
[113.]],
[[ 23.],
[123.]]],
[[[ 14.],
[114.]],
[[ 24.],
[124.]]]]]],
[[[[[[ 13.],
[113.]],
[[ 23.],
[123.]]],
[[[ 14.],
[114.]],
[[ 24.],
[124.]]],
[[[ 15.],
[115.]],
[[ 25.],
[125.]]]],
[[[[ 14.],
[114.]],
[[ 24.],
[124.]]],
[[[ 15.],
[115.]],
[[ 25.],
[125.]]],
[[[ 16.],
[116.]],
[[ 26.],
[126.]]]],
[[[[ 15.],
[115.]],
[[ 25.],
[125.]]],
[[[ 16.],
[116.]],
[[ 26.],
[126.]]],
[[[ 17.],
[117.]],
[[ 27.],
[127.]]]],
[[[[ 16.],
[116.]],
[[ 26.],
[126.]]],
[[[ 17.],
[117.]],
[[ 27.],
[127.]]],
[[[ 18.],
[118.]],
[[ 28.],
[128.]]]]],
[[[[[ 17.],
[117.]],
[[ 27.],
[127.]]],
[[[ 18.],
[118.]],
[[ 28.],
[128.]]],
[[[ 19.],
[119.]],
[[ 29.],
[129.]]]],
[[[[ 18.],
[118.]],
[[ 28.],
[128.]]],
[[[ 19.],
[119.]],
[[ 29.],
[129.]]],
[[[ 20.],
[120.]],
[[ 30.],
[130.]]]],
[[[[ 19.],
[119.]],
[[ 29.],
[129.]]],
[[[ 20.],
[120.]],
[[ 30.],
[130.]]],
[[[ 21.],
[121.]],
[[ 31.],
[131.]]]],
[[[[ 20.],
[120.]],
[[ 30.],
[130.]]],
[[[ 21.],
[121.]],
[[ 31.],
[131.]]],
[[[ 22.],
[122.]],
[[ 32.],
[132.]]]]],
[[[[[ 21.],
[121.]],
[[ 31.],
[131.]]],
[[[ 22.],
[122.]],
[[ 32.],
[132.]]],
[[[ 23.],
[123.]],
[[ 33.],
[133.]]]],
[[[[ 22.],
[122.]],
[[ 32.],
[132.]]],
[[[ 23.],
[123.]],
[[ 33.],
[133.]]],
[[[ 24.],
[124.]],
[[ 34.],
[134.]]]],
[[[[ 23.],
[123.]],
[[ 33.],
[133.]]],
[[[ 24.],
[124.]],
[[ 34.],
[134.]]],
[[[ 25.],
[125.]],
[[ 35.],
[135.]]]],
[[[[ 24.],
[124.]],
[[ 34.],
[134.]]],
[[[ 25.],
[125.]],
[[ 35.],
[135.]]],
[[[ 26.],
[126.]],
[[ 36.],
[136.]]]]]]], dtype=torch.float64)
########## torch.float32/torch.int64/size=()+(6, 6)+(2,) ##########
# sparse tensor
tensor(crow_indices=tensor([0, 2, 4]),
col_indices=tensor([0, 1, 0, 2]),
values=tensor([[[[ 1., 101.],
[ 11., 111.]],
[[ 2., 102.],
[ 12., 112.]],
[[ 3., 103.],
[ 13., 113.]]],
[[[ 2., 102.],
[ 12., 112.]],
[[ 3., 103.],
[ 13., 113.]],
[[ 4., 104.],
[ 14., 114.]]],
[[[ 3., 103.],
[ 13., 113.]],
[[ 4., 104.],
[ 14., 114.]],
[[ 5., 105.],
[ 15., 115.]]],
[[[ 4., 104.],
[ 14., 114.]],
[[ 5., 105.],
[ 15., 115.]],
[[ 6., 106.],
[ 16., 116.]]]]), size=(6, 6, 2), nnz=4,
layout=torch.sparse_bsr)
# _crow_indices
tensor([0, 2, 4])
# _col_indices
tensor([0, 1, 0, 2])
# _values
tensor([[[[ 1., 101.],
[ 11., 111.]],
[[ 2., 102.],
[ 12., 112.]],
[[ 3., 103.],
[ 13., 113.]]],
[[[ 2., 102.],
[ 12., 112.]],
[[ 3., 103.],
[ 13., 113.]],
[[ 4., 104.],
[ 14., 114.]]],
[[[ 3., 103.],
[ 13., 113.]],
[[ 4., 104.],
[ 14., 114.]],
[[ 5., 105.],
[ 15., 115.]]],
[[[ 4., 104.],
[ 14., 114.]],
[[ 5., 105.],
[ 15., 115.]],
[[ 6., 106.],
[ 16., 116.]]]])
########## torch.float32/torch.int64/size=()+(4, 9)+(4, 2) ##########
# sparse tensor
tensor(crow_indices=tensor([0, 2, 4]),
col_indices=tensor([0, 1, 0, 2]),
values=tensor([[[[[1.0000e+00, 1.0010e+03],
[1.0100e+02, 1.1010e+03],
[2.0100e+02, 1.2010e+03],
[3.0100e+02, 1.3010e+03]],
[[1.1000e+01, 1.0110e+03],
[1.1100e+02, 1.1110e+03],
[2.1100e+02, 1.2110e+03],
[3.1100e+02, 1.3110e+03]],
[[2.1000e+01, 1.0210e+03],
[1.2100e+02, 1.1210e+03],
[2.2100e+02, 1.2210e+03],
[3.2100e+02, 1.3210e+03]]],
[[[2.0000e+00, 1.0020e+03],
[1.0200e+02, 1.1020e+03],
[2.0200e+02, 1.2020e+03],
[3.0200e+02, 1.3020e+03]],
[[1.2000e+01, 1.0120e+03],
[1.1200e+02, 1.1120e+03],
[2.1200e+02, 1.2120e+03],
[3.1200e+02, 1.3120e+03]],
[[2.2000e+01, 1.0220e+03],
[1.2200e+02, 1.1220e+03],
[2.2200e+02, 1.2220e+03],
[3.2200e+02, 1.3220e+03]]]],
[[[[2.0000e+00, 1.0020e+03],
[1.0200e+02, 1.1020e+03],
[2.0200e+02, 1.2020e+03],
[3.0200e+02, 1.3020e+03]],
[[1.2000e+01, 1.0120e+03],
[1.1200e+02, 1.1120e+03],
[2.1200e+02, 1.2120e+03],
[3.1200e+02, 1.3120e+03]],
[[2.2000e+01, 1.0220e+03],
[1.2200e+02, 1.1220e+03],
[2.2200e+02, 1.2220e+03],
[3.2200e+02, 1.3220e+03]]],
[[[3.0000e+00, 1.0030e+03],
[1.0300e+02, 1.1030e+03],
[2.0300e+02, 1.2030e+03],
[3.0300e+02, 1.3030e+03]],
[[1.3000e+01, 1.0130e+03],
[1.1300e+02, 1.1130e+03],
[2.1300e+02, 1.2130e+03],
[3.1300e+02, 1.3130e+03]],
[[2.3000e+01, 1.0230e+03],
[1.2300e+02, 1.1230e+03],
[2.2300e+02, 1.2230e+03],
[3.2300e+02, 1.3230e+03]]]],
[[[[3.0000e+00, 1.0030e+03],
[1.0300e+02, 1.1030e+03],
[2.0300e+02, 1.2030e+03],
[3.0300e+02, 1.3030e+03]],
[[1.3000e+01, 1.0130e+03],
[1.1300e+02, 1.1130e+03],
[2.1300e+02, 1.2130e+03],
[3.1300e+02, 1.3130e+03]],
[[2.3000e+01, 1.0230e+03],
[1.2300e+02, 1.1230e+03],
[2.2300e+02, 1.2230e+03],
[3.2300e+02, 1.3230e+03]]],
[[[4.0000e+00, 1.0040e+03],
[1.0400e+02, 1.1040e+03],
[2.0400e+02, 1.2040e+03],
[3.0400e+02, 1.3040e+03]],
[[1.4000e+01, 1.0140e+03],
[1.1400e+02, 1.1140e+03],
[2.1400e+02, 1.2140e+03],
[3.1400e+02, 1.3140e+03]],
[[2.4000e+01, 1.0240e+03],
[1.2400e+02, 1.1240e+03],
[2.2400e+02, 1.2240e+03],
[3.2400e+02, 1.3240e+03]]]],
[[[[4.0000e+00, 1.0040e+03],
[1.0400e+02, 1.1040e+03],
[2.0400e+02, 1.2040e+03],
[3.0400e+02, 1.3040e+03]],
[[1.4000e+01, 1.0140e+03],
[1.1400e+02, 1.1140e+03],
[2.1400e+02, 1.2140e+03],
[3.1400e+02, 1.3140e+03]],
[[2.4000e+01, 1.0240e+03],
[1.2400e+02, 1.1240e+03],
[2.2400e+02, 1.2240e+03],
[3.2400e+02, 1.3240e+03]]],
[[[5.0000e+00, 1.0050e+03],
[1.0500e+02, 1.1050e+03],
[2.0500e+02, 1.2050e+03],
[3.0500e+02, 1.3050e+03]],
[[1.5000e+01, 1.0150e+03],
[1.1500e+02, 1.1150e+03],
[2.1500e+02, 1.2150e+03],
[3.1500e+02, 1.3150e+03]],
[[2.5000e+01, 1.0250e+03],
[1.2500e+02, 1.1250e+03],
[2.2500e+02, 1.2250e+03],
[3.2500e+02, 1.3250e+03]]]]]), size=(4, 9, 4, 2),
nnz=4, layout=torch.sparse_bsr)
# _crow_indices
tensor([0, 2, 4])
# _col_indices
tensor([0, 1, 0, 2])
# _values
tensor([[[[[1.0000e+00, 1.0010e+03],
[1.0100e+02, 1.1010e+03],
[2.0100e+02, 1.2010e+03],
[3.0100e+02, 1.3010e+03]],
[[1.1000e+01, 1.0110e+03],
[1.1100e+02, 1.1110e+03],
[2.1100e+02, 1.2110e+03],
[3.1100e+02, 1.3110e+03]],
[[2.1000e+01, 1.0210e+03],
[1.2100e+02, 1.1210e+03],
[2.2100e+02, 1.2210e+03],
[3.2100e+02, 1.3210e+03]]],
[[[2.0000e+00, 1.0020e+03],
[1.0200e+02, 1.1020e+03],
[2.0200e+02, 1.2020e+03],
[3.0200e+02, 1.3020e+03]],
[[1.2000e+01, 1.0120e+03],
[1.1200e+02, 1.1120e+03],
[2.1200e+02, 1.2120e+03],
[3.1200e+02, 1.3120e+03]],
[[2.2000e+01, 1.0220e+03],
[1.2200e+02, 1.1220e+03],
[2.2200e+02, 1.2220e+03],
[3.2200e+02, 1.3220e+03]]]],
[[[[2.0000e+00, 1.0020e+03],
[1.0200e+02, 1.1020e+03],
[2.0200e+02, 1.2020e+03],
[3.0200e+02, 1.3020e+03]],
[[1.2000e+01, 1.0120e+03],
[1.1200e+02, 1.1120e+03],
[2.1200e+02, 1.2120e+03],
[3.1200e+02, 1.3120e+03]],
[[2.2000e+01, 1.0220e+03],
[1.2200e+02, 1.1220e+03],
[2.2200e+02, 1.2220e+03],
[3.2200e+02, 1.3220e+03]]],
[[[3.0000e+00, 1.0030e+03],
[1.0300e+02, 1.1030e+03],
[2.0300e+02, 1.2030e+03],
[3.0300e+02, 1.3030e+03]],
[[1.3000e+01, 1.0130e+03],
[1.1300e+02, 1.1130e+03],
[2.1300e+02, 1.2130e+03],
[3.1300e+02, 1.3130e+03]],
[[2.3000e+01, 1.0230e+03],
[1.2300e+02, 1.1230e+03],
[2.2300e+02, 1.2230e+03],
[3.2300e+02, 1.3230e+03]]]],
[[[[3.0000e+00, 1.0030e+03],
[1.0300e+02, 1.1030e+03],
[2.0300e+02, 1.2030e+03],
[3.0300e+02, 1.3030e+03]],
[[1.3000e+01, 1.0130e+03],
[1.1300e+02, 1.1130e+03],
[2.1300e+02, 1.2130e+03],
[3.1300e+02, 1.3130e+03]],
[[2.3000e+01, 1.0230e+03],
[1.2300e+02, 1.1230e+03],
[2.2300e+02, 1.2230e+03],
[3.2300e+02, 1.3230e+03]]],
[[[4.0000e+00, 1.0040e+03],
[1.0400e+02, 1.1040e+03],
[2.0400e+02, 1.2040e+03],
[3.0400e+02, 1.3040e+03]],
[[1.4000e+01, 1.0140e+03],
[1.1400e+02, 1.1140e+03],
[2.1400e+02, 1.2140e+03],
[3.1400e+02, 1.3140e+03]],
[[2.4000e+01, 1.0240e+03],
[1.2400e+02, 1.1240e+03],
[2.2400e+02, 1.2240e+03],
[3.2400e+02, 1.3240e+03]]]],
[[[[4.0000e+00, 1.0040e+03],
[1.0400e+02, 1.1040e+03],
[2.0400e+02, 1.2040e+03],
[3.0400e+02, 1.3040e+03]],
[[1.4000e+01, 1.0140e+03],
[1.1400e+02, 1.1140e+03],
[2.1400e+02, 1.2140e+03],
[3.1400e+02, 1.3140e+03]],
[[2.4000e+01, 1.0240e+03],
[1.2400e+02, 1.1240e+03],
[2.2400e+02, 1.2240e+03],
[3.2400e+02, 1.3240e+03]]],
[[[5.0000e+00, 1.0050e+03],
[1.0500e+02, 1.1050e+03],
[2.0500e+02, 1.2050e+03],
[3.0500e+02, 1.3050e+03]],
[[1.5000e+01, 1.0150e+03],
[1.1500e+02, 1.1150e+03],
[2.1500e+02, 1.2150e+03],
[3.1500e+02, 1.3150e+03]],
[[2.5000e+01, 1.0250e+03],
[1.2500e+02, 1.1250e+03],
[2.2500e+02, 1.2250e+03],
[3.2500e+02, 1.3250e+03]]]]])
########## torch.float32/torch.int64/size=(2, 3)+(6, 6)+(2, 1) ##########
# sparse tensor
tensor(crow_indices=tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]]),
col_indices=tensor([[[0, 1, 0, 1],
[0, 1, 2, 0],
[0, 0, 1, 2]],
[[1, 0, 1, 2],
[0, 2, 0, 1],
[0, 1, 2, 1]]]),
values=tensor([[[[[[[ 1.],
[101.]],
[[ 11.],
[111.]]],
[[[ 2.],
[102.]],
[[ 12.],
[112.]]],
[[[ 3.],
[103.]],
[[ 13.],
[113.]]]],
[[[[ 2.],
[102.]],
[[ 12.],
[112.]]],
[[[ 3.],
[103.]],
[[ 13.],
[113.]]],
[[[ 4.],
[104.]],
[[ 14.],
[114.]]]],
[[[[ 3.],
[103.]],
[[ 13.],
[113.]]],
[[[ 4.],
[104.]],
[[ 14.],
[114.]]],
[[[ 5.],
[105.]],
[[ 15.],
[115.]]]],
[[[[ 4.],
[104.]],
[[ 14.],
[114.]]],
[[[ 5.],
[105.]],
[[ 15.],
[115.]]],
[[[ 6.],
[106.]],
[[ 16.],
[116.]]]]],
[[[[[ 5.],
[105.]],
[[ 15.],
[115.]]],
[[[ 6.],
[106.]],
[[ 16.],
[116.]]],
[[[ 7.],
[107.]],
[[ 17.],
[117.]]]],
[[[[ 6.],
[106.]],
[[ 16.],
[116.]]],
[[[ 7.],
[107.]],
[[ 17.],
[117.]]],
[[[ 8.],
[108.]],
[[ 18.],
[118.]]]],
[[[[ 7.],
[107.]],
[[ 17.],
[117.]]],
[[[ 8.],
[108.]],
[[ 18.],
[118.]]],
[[[ 9.],
[109.]],
[[ 19.],
[119.]]]],
[[[[ 8.],
[108.]],
[[ 18.],
[118.]]],
[[[ 9.],
[109.]],
[[ 19.],
[119.]]],
[[[ 10.],
[110.]],
[[ 20.],
[120.]]]]],
[[[[[ 9.],
[109.]],
[[ 19.],
[119.]]],
[[[ 10.],
[110.]],
[[ 20.],
[120.]]],
[[[ 11.],
[111.]],
[[ 21.],
[121.]]]],
[[[[ 10.],
[110.]],
[[ 20.],
[120.]]],
[[[ 11.],
[111.]],
[[ 21.],
[121.]]],
[[[ 12.],
[112.]],
[[ 22.],
[122.]]]],
[[[[ 11.],
[111.]],
[[ 21.],
[121.]]],
[[[ 12.],
[112.]],
[[ 22.],
[122.]]],
[[[ 13.],
[113.]],
[[ 23.],
[123.]]]],
[[[[ 12.],
[112.]],
[[ 22.],
[122.]]],
[[[ 13.],
[113.]],
[[ 23.],
[123.]]],
[[[ 14.],
[114.]],
[[ 24.],
[124.]]]]]],
[[[[[[ 13.],
[113.]],
[[ 23.],
[123.]]],
[[[ 14.],
[114.]],
[[ 24.],
[124.]]],
[[[ 15.],
[115.]],
[[ 25.],
[125.]]]],
[[[[ 14.],
[114.]],
[[ 24.],
[124.]]],
[[[ 15.],
[115.]],
[[ 25.],
[125.]]],
[[[ 16.],
[116.]],
[[ 26.],
[126.]]]],
[[[[ 15.],
[115.]],
[[ 25.],
[125.]]],
[[[ 16.],
[116.]],
[[ 26.],
[126.]]],
[[[ 17.],
[117.]],
[[ 27.],
[127.]]]],
[[[[ 16.],
[116.]],
[[ 26.],
[126.]]],
[[[ 17.],
[117.]],
[[ 27.],
[127.]]],
[[[ 18.],
[118.]],
[[ 28.],
[128.]]]]],
[[[[[ 17.],
[117.]],
[[ 27.],
[127.]]],
[[[ 18.],
[118.]],
[[ 28.],
[128.]]],
[[[ 19.],
[119.]],
[[ 29.],
[129.]]]],
[[[[ 18.],
[118.]],
[[ 28.],
[128.]]],
[[[ 19.],
[119.]],
[[ 29.],
[129.]]],
[[[ 20.],
[120.]],
[[ 30.],
[130.]]]],
[[[[ 19.],
[119.]],
[[ 29.],
[129.]]],
[[[ 20.],
[120.]],
[[ 30.],
[130.]]],
[[[ 21.],
[121.]],
[[ 31.],
[131.]]]],
[[[[ 20.],
[120.]],
[[ 30.],
[130.]]],
[[[ 21.],
[121.]],
[[ 31.],
[131.]]],
[[[ 22.],
[122.]],
[[ 32.],
[132.]]]]],
[[[[[ 21.],
[121.]],
[[ 31.],
[131.]]],
[[[ 22.],
[122.]],
[[ 32.],
[132.]]],
[[[ 23.],
[123.]],
[[ 33.],
[133.]]]],
[[[[ 22.],
[122.]],
[[ 32.],
[132.]]],
[[[ 23.],
[123.]],
[[ 33.],
[133.]]],
[[[ 24.],
[124.]],
[[ 34.],
[134.]]]],
[[[[ 23.],
[123.]],
[[ 33.],
[133.]]],
[[[ 24.],
[124.]],
[[ 34.],
[134.]]],
[[[ 25.],
[125.]],
[[ 35.],
[135.]]]],
[[[[ 24.],
[124.]],
[[ 34.],
[134.]]],
[[[ 25.],
[125.]],
[[ 35.],
[135.]]],
[[[ 26.],
[126.]],
[[ 36.],
[136.]]]]]]]), size=(2, 3, 6, 6, 2, 1), nnz=4,
layout=torch.sparse_bsr)
# _crow_indices
tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]])
# _col_indices
tensor([[[0, 1, 0, 1],
[0, 1, 2, 0],
[0, 0, 1, 2]],
[[1, 0, 1, 2],
[0, 2, 0, 1],
[0, 1, 2, 1]]])
# _values
tensor([[[[[[[ 1.],
[101.]],
[[ 11.],
[111.]]],
[[[ 2.],
[102.]],
[[ 12.],
[112.]]],
[[[ 3.],
[103.]],
[[ 13.],
[113.]]]],
[[[[ 2.],
[102.]],
[[ 12.],
[112.]]],
[[[ 3.],
[103.]],
[[ 13.],
[113.]]],
[[[ 4.],
[104.]],
[[ 14.],
[114.]]]],
[[[[ 3.],
[103.]],
[[ 13.],
[113.]]],
[[[ 4.],
[104.]],
[[ 14.],
[114.]]],
[[[ 5.],
[105.]],
[[ 15.],
[115.]]]],
[[[[ 4.],
[104.]],
[[ 14.],
[114.]]],
[[[ 5.],
[105.]],
[[ 15.],
[115.]]],
[[[ 6.],
[106.]],
[[ 16.],
[116.]]]]],
[[[[[ 5.],
[105.]],
[[ 15.],
[115.]]],
[[[ 6.],
[106.]],
[[ 16.],
[116.]]],
[[[ 7.],
[107.]],
[[ 17.],
[117.]]]],
[[[[ 6.],
[106.]],
[[ 16.],
[116.]]],
[[[ 7.],
[107.]],
[[ 17.],
[117.]]],
[[[ 8.],
[108.]],
[[ 18.],
[118.]]]],
[[[[ 7.],
[107.]],
[[ 17.],
[117.]]],
[[[ 8.],
[108.]],
[[ 18.],
[118.]]],
[[[ 9.],
[109.]],
[[ 19.],
[119.]]]],
[[[[ 8.],
[108.]],
[[ 18.],
[118.]]],
[[[ 9.],
[109.]],
[[ 19.],
[119.]]],
[[[ 10.],
[110.]],
[[ 20.],
[120.]]]]],
[[[[[ 9.],
[109.]],
[[ 19.],
[119.]]],
[[[ 10.],
[110.]],
[[ 20.],
[120.]]],
[[[ 11.],
[111.]],
[[ 21.],
[121.]]]],
[[[[ 10.],
[110.]],
[[ 20.],
[120.]]],
[[[ 11.],
[111.]],
[[ 21.],
[121.]]],
[[[ 12.],
[112.]],
[[ 22.],
[122.]]]],
[[[[ 11.],
[111.]],
[[ 21.],
[121.]]],
[[[ 12.],
[112.]],
[[ 22.],
[122.]]],
[[[ 13.],
[113.]],
[[ 23.],
[123.]]]],
[[[[ 12.],
[112.]],
[[ 22.],
[122.]]],
[[[ 13.],
[113.]],
[[ 23.],
[123.]]],
[[[ 14.],
[114.]],
[[ 24.],
[124.]]]]]],
[[[[[[ 13.],
[113.]],
[[ 23.],
[123.]]],
[[[ 14.],
[114.]],
[[ 24.],
[124.]]],
[[[ 15.],
[115.]],
[[ 25.],
[125.]]]],
[[[[ 14.],
[114.]],
[[ 24.],
[124.]]],
[[[ 15.],
[115.]],
[[ 25.],
[125.]]],
[[[ 16.],
[116.]],
[[ 26.],
[126.]]]],
[[[[ 15.],
[115.]],
[[ 25.],
[125.]]],
[[[ 16.],
[116.]],
[[ 26.],
[126.]]],
[[[ 17.],
[117.]],
[[ 27.],
[127.]]]],
[[[[ 16.],
[116.]],
[[ 26.],
[126.]]],
[[[ 17.],
[117.]],
[[ 27.],
[127.]]],
[[[ 18.],
[118.]],
[[ 28.],
[128.]]]]],
[[[[[ 17.],
[117.]],
[[ 27.],
[127.]]],
[[[ 18.],
[118.]],
[[ 28.],
[128.]]],
[[[ 19.],
[119.]],
[[ 29.],
[129.]]]],
[[[[ 18.],
[118.]],
[[ 28.],
[128.]]],
[[[ 19.],
[119.]],
[[ 29.],
[129.]]],
[[[ 20.],
[120.]],
[[ 30.],
[130.]]]],
[[[[ 19.],
[119.]],
[[ 29.],
[129.]]],
[[[ 20.],
[120.]],
[[ 30.],
[130.]]],
[[[ 21.],
[121.]],
[[ 31.],
[131.]]]],
[[[[ 20.],
[120.]],
[[ 30.],
[130.]]],
[[[ 21.],
[121.]],
[[ 31.],
[131.]]],
[[[ 22.],
[122.]],
[[ 32.],
[132.]]]]],
[[[[[ 21.],
[121.]],
[[ 31.],
[131.]]],
[[[ 22.],
[122.]],
[[ 32.],
[132.]]],
[[[ 23.],
[123.]],
[[ 33.],
[133.]]]],
[[[[ 22.],
[122.]],
[[ 32.],
[132.]]],
[[[ 23.],
[123.]],
[[ 33.],
[133.]]],
[[[ 24.],
[124.]],
[[ 34.],
[134.]]]],
[[[[ 23.],
[123.]],
[[ 33.],
[133.]]],
[[[ 24.],
[124.]],
[[ 34.],
[134.]]],
[[[ 25.],
[125.]],
[[ 35.],
[135.]]]],
[[[[ 24.],
[124.]],
[[ 34.],
[134.]]],
[[[ 25.],
[125.]],
[[ 35.],
[135.]]],
[[[ 26.],
[126.]],
[[ 36.],
[136.]]]]]]])
########## torch.float64/torch.int64/size=()+(6, 6)+(2,) ##########
# sparse tensor
tensor(crow_indices=tensor([0, 2, 4]),
col_indices=tensor([0, 1, 0, 2]),
values=tensor([[[[ 1., 101.],
[ 11., 111.]],
[[ 2., 102.],
[ 12., 112.]],
[[ 3., 103.],
[ 13., 113.]]],
[[[ 2., 102.],
[ 12., 112.]],
[[ 3., 103.],
[ 13., 113.]],
[[ 4., 104.],
[ 14., 114.]]],
[[[ 3., 103.],
[ 13., 113.]],
[[ 4., 104.],
[ 14., 114.]],
[[ 5., 105.],
[ 15., 115.]]],
[[[ 4., 104.],
[ 14., 114.]],
[[ 5., 105.],
[ 15., 115.]],
[[ 6., 106.],
[ 16., 116.]]]]), size=(6, 6, 2), nnz=4,
dtype=torch.float64, layout=torch.sparse_bsr)
# _crow_indices
tensor([0, 2, 4])
# _col_indices
tensor([0, 1, 0, 2])
# _values
tensor([[[[ 1., 101.],
[ 11., 111.]],
[[ 2., 102.],
[ 12., 112.]],
[[ 3., 103.],
[ 13., 113.]]],
[[[ 2., 102.],
[ 12., 112.]],
[[ 3., 103.],
[ 13., 113.]],
[[ 4., 104.],
[ 14., 114.]]],
[[[ 3., 103.],
[ 13., 113.]],
[[ 4., 104.],
[ 14., 114.]],
[[ 5., 105.],
[ 15., 115.]]],
[[[ 4., 104.],
[ 14., 114.]],
[[ 5., 105.],
[ 15., 115.]],
[[ 6., 106.],
[ 16., 116.]]]], dtype=torch.float64)
########## torch.float64/torch.int64/size=()+(4, 9)+(4, 2) ##########
# sparse tensor
tensor(crow_indices=tensor([0, 2, 4]),
col_indices=tensor([0, 1, 0, 2]),
values=tensor([[[[[1.0000e+00, 1.0010e+03],
[1.0100e+02, 1.1010e+03],
[2.0100e+02, 1.2010e+03],
[3.0100e+02, 1.3010e+03]],
[[1.1000e+01, 1.0110e+03],
[1.1100e+02, 1.1110e+03],
[2.1100e+02, 1.2110e+03],
[3.1100e+02, 1.3110e+03]],
[[2.1000e+01, 1.0210e+03],
[1.2100e+02, 1.1210e+03],
[2.2100e+02, 1.2210e+03],
[3.2100e+02, 1.3210e+03]]],
[[[2.0000e+00, 1.0020e+03],
[1.0200e+02, 1.1020e+03],
[2.0200e+02, 1.2020e+03],
[3.0200e+02, 1.3020e+03]],
[[1.2000e+01, 1.0120e+03],
[1.1200e+02, 1.1120e+03],
[2.1200e+02, 1.2120e+03],
[3.1200e+02, 1.3120e+03]],
[[2.2000e+01, 1.0220e+03],
[1.2200e+02, 1.1220e+03],
[2.2200e+02, 1.2220e+03],
[3.2200e+02, 1.3220e+03]]]],
[[[[2.0000e+00, 1.0020e+03],
[1.0200e+02, 1.1020e+03],
[2.0200e+02, 1.2020e+03],
[3.0200e+02, 1.3020e+03]],
[[1.2000e+01, 1.0120e+03],
[1.1200e+02, 1.1120e+03],
[2.1200e+02, 1.2120e+03],
[3.1200e+02, 1.3120e+03]],
[[2.2000e+01, 1.0220e+03],
[1.2200e+02, 1.1220e+03],
[2.2200e+02, 1.2220e+03],
[3.2200e+02, 1.3220e+03]]],
[[[3.0000e+00, 1.0030e+03],
[1.0300e+02, 1.1030e+03],
[2.0300e+02, 1.2030e+03],
[3.0300e+02, 1.3030e+03]],
[[1.3000e+01, 1.0130e+03],
[1.1300e+02, 1.1130e+03],
[2.1300e+02, 1.2130e+03],
[3.1300e+02, 1.3130e+03]],
[[2.3000e+01, 1.0230e+03],
[1.2300e+02, 1.1230e+03],
[2.2300e+02, 1.2230e+03],
[3.2300e+02, 1.3230e+03]]]],
[[[[3.0000e+00, 1.0030e+03],
[1.0300e+02, 1.1030e+03],
[2.0300e+02, 1.2030e+03],
[3.0300e+02, 1.3030e+03]],
[[1.3000e+01, 1.0130e+03],
[1.1300e+02, 1.1130e+03],
[2.1300e+02, 1.2130e+03],
[3.1300e+02, 1.3130e+03]],
[[2.3000e+01, 1.0230e+03],
[1.2300e+02, 1.1230e+03],
[2.2300e+02, 1.2230e+03],
[3.2300e+02, 1.3230e+03]]],
[[[4.0000e+00, 1.0040e+03],
[1.0400e+02, 1.1040e+03],
[2.0400e+02, 1.2040e+03],
[3.0400e+02, 1.3040e+03]],
[[1.4000e+01, 1.0140e+03],
[1.1400e+02, 1.1140e+03],
[2.1400e+02, 1.2140e+03],
[3.1400e+02, 1.3140e+03]],
[[2.4000e+01, 1.0240e+03],
[1.2400e+02, 1.1240e+03],
[2.2400e+02, 1.2240e+03],
[3.2400e+02, 1.3240e+03]]]],
[[[[4.0000e+00, 1.0040e+03],
[1.0400e+02, 1.1040e+03],
[2.0400e+02, 1.2040e+03],
[3.0400e+02, 1.3040e+03]],
[[1.4000e+01, 1.0140e+03],
[1.1400e+02, 1.1140e+03],
[2.1400e+02, 1.2140e+03],
[3.1400e+02, 1.3140e+03]],
[[2.4000e+01, 1.0240e+03],
[1.2400e+02, 1.1240e+03],
[2.2400e+02, 1.2240e+03],
[3.2400e+02, 1.3240e+03]]],
[[[5.0000e+00, 1.0050e+03],
[1.0500e+02, 1.1050e+03],
[2.0500e+02, 1.2050e+03],
[3.0500e+02, 1.3050e+03]],
[[1.5000e+01, 1.0150e+03],
[1.1500e+02, 1.1150e+03],
[2.1500e+02, 1.2150e+03],
[3.1500e+02, 1.3150e+03]],
[[2.5000e+01, 1.0250e+03],
[1.2500e+02, 1.1250e+03],
[2.2500e+02, 1.2250e+03],
[3.2500e+02, 1.3250e+03]]]]]), size=(4, 9, 4, 2),
nnz=4, dtype=torch.float64, layout=torch.sparse_bsr)
# _crow_indices
tensor([0, 2, 4])
# _col_indices
tensor([0, 1, 0, 2])
# _values
tensor([[[[[1.0000e+00, 1.0010e+03],
[1.0100e+02, 1.1010e+03],
[2.0100e+02, 1.2010e+03],
[3.0100e+02, 1.3010e+03]],
[[1.1000e+01, 1.0110e+03],
[1.1100e+02, 1.1110e+03],
[2.1100e+02, 1.2110e+03],
[3.1100e+02, 1.3110e+03]],
[[2.1000e+01, 1.0210e+03],
[1.2100e+02, 1.1210e+03],
[2.2100e+02, 1.2210e+03],
[3.2100e+02, 1.3210e+03]]],
[[[2.0000e+00, 1.0020e+03],
[1.0200e+02, 1.1020e+03],
[2.0200e+02, 1.2020e+03],
[3.0200e+02, 1.3020e+03]],
[[1.2000e+01, 1.0120e+03],
[1.1200e+02, 1.1120e+03],
[2.1200e+02, 1.2120e+03],
[3.1200e+02, 1.3120e+03]],
[[2.2000e+01, 1.0220e+03],
[1.2200e+02, 1.1220e+03],
[2.2200e+02, 1.2220e+03],
[3.2200e+02, 1.3220e+03]]]],
[[[[2.0000e+00, 1.0020e+03],
[1.0200e+02, 1.1020e+03],
[2.0200e+02, 1.2020e+03],
[3.0200e+02, 1.3020e+03]],
[[1.2000e+01, 1.0120e+03],
[1.1200e+02, 1.1120e+03],
[2.1200e+02, 1.2120e+03],
[3.1200e+02, 1.3120e+03]],
[[2.2000e+01, 1.0220e+03],
[1.2200e+02, 1.1220e+03],
[2.2200e+02, 1.2220e+03],
[3.2200e+02, 1.3220e+03]]],
[[[3.0000e+00, 1.0030e+03],
[1.0300e+02, 1.1030e+03],
[2.0300e+02, 1.2030e+03],
[3.0300e+02, 1.3030e+03]],
[[1.3000e+01, 1.0130e+03],
[1.1300e+02, 1.1130e+03],
[2.1300e+02, 1.2130e+03],
[3.1300e+02, 1.3130e+03]],
[[2.3000e+01, 1.0230e+03],
[1.2300e+02, 1.1230e+03],
[2.2300e+02, 1.2230e+03],
[3.2300e+02, 1.3230e+03]]]],
[[[[3.0000e+00, 1.0030e+03],
[1.0300e+02, 1.1030e+03],
[2.0300e+02, 1.2030e+03],
[3.0300e+02, 1.3030e+03]],
[[1.3000e+01, 1.0130e+03],
[1.1300e+02, 1.1130e+03],
[2.1300e+02, 1.2130e+03],
[3.1300e+02, 1.3130e+03]],
[[2.3000e+01, 1.0230e+03],
[1.2300e+02, 1.1230e+03],
[2.2300e+02, 1.2230e+03],
[3.2300e+02, 1.3230e+03]]],
[[[4.0000e+00, 1.0040e+03],
[1.0400e+02, 1.1040e+03],
[2.0400e+02, 1.2040e+03],
[3.0400e+02, 1.3040e+03]],
[[1.4000e+01, 1.0140e+03],
[1.1400e+02, 1.1140e+03],
[2.1400e+02, 1.2140e+03],
[3.1400e+02, 1.3140e+03]],
[[2.4000e+01, 1.0240e+03],
[1.2400e+02, 1.1240e+03],
[2.2400e+02, 1.2240e+03],
[3.2400e+02, 1.3240e+03]]]],
[[[[4.0000e+00, 1.0040e+03],
[1.0400e+02, 1.1040e+03],
[2.0400e+02, 1.2040e+03],
[3.0400e+02, 1.3040e+03]],
[[1.4000e+01, 1.0140e+03],
[1.1400e+02, 1.1140e+03],
[2.1400e+02, 1.2140e+03],
[3.1400e+02, 1.3140e+03]],
[[2.4000e+01, 1.0240e+03],
[1.2400e+02, 1.1240e+03],
[2.2400e+02, 1.2240e+03],
[3.2400e+02, 1.3240e+03]]],
[[[5.0000e+00, 1.0050e+03],
[1.0500e+02, 1.1050e+03],
[2.0500e+02, 1.2050e+03],
[3.0500e+02, 1.3050e+03]],
[[1.5000e+01, 1.0150e+03],
[1.1500e+02, 1.1150e+03],
[2.1500e+02, 1.2150e+03],
[3.1500e+02, 1.3150e+03]],
[[2.5000e+01, 1.0250e+03],
[1.2500e+02, 1.1250e+03],
[2.2500e+02, 1.2250e+03],
[3.2500e+02, 1.3250e+03]]]]], dtype=torch.float64)
########## torch.float64/torch.int64/size=(2, 3)+(6, 6)+(2, 1) ##########
# sparse tensor
tensor(crow_indices=tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]]),
col_indices=tensor([[[0, 1, 0, 1],
[0, 1, 2, 0],
[0, 0, 1, 2]],
[[1, 0, 1, 2],
[0, 2, 0, 1],
[0, 1, 2, 1]]]),
values=tensor([[[[[[[ 1.],
[101.]],
[[ 11.],
[111.]]],
[[[ 2.],
[102.]],
[[ 12.],
[112.]]],
[[[ 3.],
[103.]],
[[ 13.],
[113.]]]],
[[[[ 2.],
[102.]],
[[ 12.],
[112.]]],
[[[ 3.],
[103.]],
[[ 13.],
[113.]]],
[[[ 4.],
[104.]],
[[ 14.],
[114.]]]],
[[[[ 3.],
[103.]],
[[ 13.],
[113.]]],
[[[ 4.],
[104.]],
[[ 14.],
[114.]]],
[[[ 5.],
[105.]],
[[ 15.],
[115.]]]],
[[[[ 4.],
[104.]],
[[ 14.],
[114.]]],
[[[ 5.],
[105.]],
[[ 15.],
[115.]]],
[[[ 6.],
[106.]],
[[ 16.],
[116.]]]]],
[[[[[ 5.],
[105.]],
[[ 15.],
[115.]]],
[[[ 6.],
[106.]],
[[ 16.],
[116.]]],
[[[ 7.],
[107.]],
[[ 17.],
[117.]]]],
[[[[ 6.],
[106.]],
[[ 16.],
[116.]]],
[[[ 7.],
[107.]],
[[ 17.],
[117.]]],
[[[ 8.],
[108.]],
[[ 18.],
[118.]]]],
[[[[ 7.],
[107.]],
[[ 17.],
[117.]]],
[[[ 8.],
[108.]],
[[ 18.],
[118.]]],
[[[ 9.],
[109.]],
[[ 19.],
[119.]]]],
[[[[ 8.],
[108.]],
[[ 18.],
[118.]]],
[[[ 9.],
[109.]],
[[ 19.],
[119.]]],
[[[ 10.],
[110.]],
[[ 20.],
[120.]]]]],
[[[[[ 9.],
[109.]],
[[ 19.],
[119.]]],
[[[ 10.],
[110.]],
[[ 20.],
[120.]]],
[[[ 11.],
[111.]],
[[ 21.],
[121.]]]],
[[[[ 10.],
[110.]],
[[ 20.],
[120.]]],
[[[ 11.],
[111.]],
[[ 21.],
[121.]]],
[[[ 12.],
[112.]],
[[ 22.],
[122.]]]],
[[[[ 11.],
[111.]],
[[ 21.],
[121.]]],
[[[ 12.],
[112.]],
[[ 22.],
[122.]]],
[[[ 13.],
[113.]],
[[ 23.],
[123.]]]],
[[[[ 12.],
[112.]],
[[ 22.],
[122.]]],
[[[ 13.],
[113.]],
[[ 23.],
[123.]]],
[[[ 14.],
[114.]],
[[ 24.],
[124.]]]]]],
[[[[[[ 13.],
[113.]],
[[ 23.],
[123.]]],
[[[ 14.],
[114.]],
[[ 24.],
[124.]]],
[[[ 15.],
[115.]],
[[ 25.],
[125.]]]],
[[[[ 14.],
[114.]],
[[ 24.],
[124.]]],
[[[ 15.],
[115.]],
[[ 25.],
[125.]]],
[[[ 16.],
[116.]],
[[ 26.],
[126.]]]],
[[[[ 15.],
[115.]],
[[ 25.],
[125.]]],
[[[ 16.],
[116.]],
[[ 26.],
[126.]]],
[[[ 17.],
[117.]],
[[ 27.],
[127.]]]],
[[[[ 16.],
[116.]],
[[ 26.],
[126.]]],
[[[ 17.],
[117.]],
[[ 27.],
[127.]]],
[[[ 18.],
[118.]],
[[ 28.],
[128.]]]]],
[[[[[ 17.],
[117.]],
[[ 27.],
[127.]]],
[[[ 18.],
[118.]],
[[ 28.],
[128.]]],
[[[ 19.],
[119.]],
[[ 29.],
[129.]]]],
[[[[ 18.],
[118.]],
[[ 28.],
[128.]]],
[[[ 19.],
[119.]],
[[ 29.],
[129.]]],
[[[ 20.],
[120.]],
[[ 30.],
[130.]]]],
[[[[ 19.],
[119.]],
[[ 29.],
[129.]]],
[[[ 20.],
[120.]],
[[ 30.],
[130.]]],
[[[ 21.],
[121.]],
[[ 31.],
[131.]]]],
[[[[ 20.],
[120.]],
[[ 30.],
[130.]]],
[[[ 21.],
[121.]],
[[ 31.],
[131.]]],
[[[ 22.],
[122.]],
[[ 32.],
[132.]]]]],
[[[[[ 21.],
[121.]],
[[ 31.],
[131.]]],
[[[ 22.],
[122.]],
[[ 32.],
[132.]]],
[[[ 23.],
[123.]],
[[ 33.],
[133.]]]],
[[[[ 22.],
[122.]],
[[ 32.],
[132.]]],
[[[ 23.],
[123.]],
[[ 33.],
[133.]]],
[[[ 24.],
[124.]],
[[ 34.],
[134.]]]],
[[[[ 23.],
[123.]],
[[ 33.],
[133.]]],
[[[ 24.],
[124.]],
[[ 34.],
[134.]]],
[[[ 25.],
[125.]],
[[ 35.],
[135.]]]],
[[[[ 24.],
[124.]],
[[ 34.],
[134.]]],
[[[ 25.],
[125.]],
[[ 35.],
[135.]]],
[[[ 26.],
[126.]],
[[ 36.],
[136.]]]]]]]), size=(2, 3, 6, 6, 2, 1), nnz=4,
dtype=torch.float64, layout=torch.sparse_bsr)
# _crow_indices
tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]])
# _col_indices
tensor([[[0, 1, 0, 1],
[0, 1, 2, 0],
[0, 0, 1, 2]],
[[1, 0, 1, 2],
[0, 2, 0, 1],
[0, 1, 2, 1]]])
# _values
tensor([[[[[[[ 1.],
[101.]],
[[ 11.],
[111.]]],
[[[ 2.],
[102.]],
[[ 12.],
[112.]]],
[[[ 3.],
[103.]],
[[ 13.],
[113.]]]],
[[[[ 2.],
[102.]],
[[ 12.],
[112.]]],
[[[ 3.],
[103.]],
[[ 13.],
[113.]]],
[[[ 4.],
[104.]],
[[ 14.],
[114.]]]],
[[[[ 3.],
[103.]],
[[ 13.],
[113.]]],
[[[ 4.],
[104.]],
[[ 14.],
[114.]]],
[[[ 5.],
[105.]],
[[ 15.],
[115.]]]],
[[[[ 4.],
[104.]],
[[ 14.],
[114.]]],
[[[ 5.],
[105.]],
[[ 15.],
[115.]]],
[[[ 6.],
[106.]],
[[ 16.],
[116.]]]]],
[[[[[ 5.],
[105.]],
[[ 15.],
[115.]]],
[[[ 6.],
[106.]],
[[ 16.],
[116.]]],
[[[ 7.],
[107.]],
[[ 17.],
[117.]]]],
[[[[ 6.],
[106.]],
[[ 16.],
[116.]]],
[[[ 7.],
[107.]],
[[ 17.],
[117.]]],
[[[ 8.],
[108.]],
[[ 18.],
[118.]]]],
[[[[ 7.],
[107.]],
[[ 17.],
[117.]]],
[[[ 8.],
[108.]],
[[ 18.],
[118.]]],
[[[ 9.],
[109.]],
[[ 19.],
[119.]]]],
[[[[ 8.],
[108.]],
[[ 18.],
[118.]]],
[[[ 9.],
[109.]],
[[ 19.],
[119.]]],
[[[ 10.],
[110.]],
[[ 20.],
[120.]]]]],
[[[[[ 9.],
[109.]],
[[ 19.],
[119.]]],
[[[ 10.],
[110.]],
[[ 20.],
[120.]]],
[[[ 11.],
[111.]],
[[ 21.],
[121.]]]],
[[[[ 10.],
[110.]],
[[ 20.],
[120.]]],
[[[ 11.],
[111.]],
[[ 21.],
[121.]]],
[[[ 12.],
[112.]],
[[ 22.],
[122.]]]],
[[[[ 11.],
[111.]],
[[ 21.],
[121.]]],
[[[ 12.],
[112.]],
[[ 22.],
[122.]]],
[[[ 13.],
[113.]],
[[ 23.],
[123.]]]],
[[[[ 12.],
[112.]],
[[ 22.],
[122.]]],
[[[ 13.],
[113.]],
[[ 23.],
[123.]]],
[[[ 14.],
[114.]],
[[ 24.],
[124.]]]]]],
[[[[[[ 13.],
[113.]],
[[ 23.],
[123.]]],
[[[ 14.],
[114.]],
[[ 24.],
[124.]]],
[[[ 15.],
[115.]],
[[ 25.],
[125.]]]],
[[[[ 14.],
[114.]],
[[ 24.],
[124.]]],
[[[ 15.],
[115.]],
[[ 25.],
[125.]]],
[[[ 16.],
[116.]],
[[ 26.],
[126.]]]],
[[[[ 15.],
[115.]],
[[ 25.],
[125.]]],
[[[ 16.],
[116.]],
[[ 26.],
[126.]]],
[[[ 17.],
[117.]],
[[ 27.],
[127.]]]],
[[[[ 16.],
[116.]],
[[ 26.],
[126.]]],
[[[ 17.],
[117.]],
[[ 27.],
[127.]]],
[[[ 18.],
[118.]],
[[ 28.],
[128.]]]]],
[[[[[ 17.],
[117.]],
[[ 27.],
[127.]]],
[[[ 18.],
[118.]],
[[ 28.],
[128.]]],
[[[ 19.],
[119.]],
[[ 29.],
[129.]]]],
[[[[ 18.],
[118.]],
[[ 28.],
[128.]]],
[[[ 19.],
[119.]],
[[ 29.],
[129.]]],
[[[ 20.],
[120.]],
[[ 30.],
[130.]]]],
[[[[ 19.],
[119.]],
[[ 29.],
[129.]]],
[[[ 20.],
[120.]],
[[ 30.],
[130.]]],
[[[ 21.],
[121.]],
[[ 31.],
[131.]]]],
[[[[ 20.],
[120.]],
[[ 30.],
[130.]]],
[[[ 21.],
[121.]],
[[ 31.],
[131.]]],
[[[ 22.],
[122.]],
[[ 32.],
[132.]]]]],
[[[[[ 21.],
[121.]],
[[ 31.],
[131.]]],
[[[ 22.],
[122.]],
[[ 32.],
[132.]]],
[[[ 23.],
[123.]],
[[ 33.],
[133.]]]],
[[[[ 22.],
[122.]],
[[ 32.],
[132.]]],
[[[ 23.],
[123.]],
[[ 33.],
[133.]]],
[[[ 24.],
[124.]],
[[ 34.],
[134.]]]],
[[[[ 23.],
[123.]],
[[ 33.],
[133.]]],
[[[ 24.],
[124.]],
[[ 34.],
[134.]]],
[[[ 25.],
[125.]],
[[ 35.],
[135.]]]],
[[[[ 24.],
[124.]],
[[ 34.],
[134.]]],
[[[ 25.],
[125.]],
[[ 35.],
[135.]]],
[[[ 26.],
[126.]],
[[ 36.],
[136.]]]]]]], dtype=torch.float64)