Google Git
Sign in
android / platform / external / pytorch / d0e12d1cc8b08ea8770b6ec941372793c4e4d4d0 / . / test / expect / TestSparseCompressedCPU.test_print_SparseCSR_cpu.expect
blob: 3ab2e1135aa553996a52a3d62653eb40e836f456 [file] [log] [blame]
########## torch.float32/torch.int32/size=()+(2, 3)+() ##########
# sparse tensor
tensor(crow_indices=tensor([0, 2, 4]),
col_indices=tensor([0, 1, 0, 2]),
values=tensor([1., 2., 3., 4.]), size=(2, 3), nnz=4,
layout=torch.sparse_csr)
# _crow_indices
tensor([0, 2, 4], dtype=torch.int32)
# _col_indices
tensor([0, 1, 0, 2], dtype=torch.int32)
# _values
tensor([1., 2., 3., 4.])
########## torch.float32/torch.int32/size=()+(0, 0)+() ##########
# sparse tensor
tensor(crow_indices=tensor([0]),
col_indices=tensor([], size=(0,)),
values=tensor([], size=(0,)), size=(0, 0), nnz=0,
layout=torch.sparse_csr)
# _crow_indices
tensor([0], dtype=torch.int32)
# _col_indices
tensor([], dtype=torch.int32)
# _values
tensor([])
########## torch.float32/torch.int32/size=(2,)+(2, 3)+() ##########
# 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., 2., 3., 4.],
[5., 6., 7., 8.]]), size=(2, 2, 3), nnz=4,
layout=torch.sparse_csr)
# _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., 2., 3., 4.],
[5., 6., 7., 8.]])
########## torch.float32/torch.int32/size=(2, 3)+(2, 3)+() ##########
# 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., 2., 3., 4.],
[ 5., 6., 7., 8.],
[ 9., 10., 11., 12.]],
[[13., 14., 15., 16.],
[17., 18., 19., 20.],
[21., 22., 23., 24.]]]), size=(2, 3, 2, 3), nnz=4,
layout=torch.sparse_csr)
# _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., 2., 3., 4.],
[ 5., 6., 7., 8.],
[ 9., 10., 11., 12.]],
[[13., 14., 15., 16.],
[17., 18., 19., 20.],
[21., 22., 23., 24.]]])
########## torch.float64/torch.int32/size=()+(2, 3)+() ##########
# sparse tensor
tensor(crow_indices=tensor([0, 2, 4]),
col_indices=tensor([0, 1, 0, 2]),
values=tensor([1., 2., 3., 4.]), size=(2, 3), nnz=4,
dtype=torch.float64, layout=torch.sparse_csr)
# _crow_indices
tensor([0, 2, 4], dtype=torch.int32)
# _col_indices
tensor([0, 1, 0, 2], dtype=torch.int32)
# _values
tensor([1., 2., 3., 4.], dtype=torch.float64)
########## torch.float64/torch.int32/size=()+(0, 0)+() ##########
# sparse tensor
tensor(crow_indices=tensor([0]),
col_indices=tensor([], size=(0,)),
values=tensor([], size=(0,)), size=(0, 0), nnz=0, dtype=torch.float64,
layout=torch.sparse_csr)
# _crow_indices
tensor([0], dtype=torch.int32)
# _col_indices
tensor([], dtype=torch.int32)
# _values
tensor([], dtype=torch.float64)
########## torch.float64/torch.int32/size=(2,)+(2, 3)+() ##########
# 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., 2., 3., 4.],
[5., 6., 7., 8.]]), size=(2, 2, 3), nnz=4,
dtype=torch.float64, layout=torch.sparse_csr)
# _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., 2., 3., 4.],
[5., 6., 7., 8.]], dtype=torch.float64)
########## torch.float64/torch.int32/size=(2, 3)+(2, 3)+() ##########
# 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., 2., 3., 4.],
[ 5., 6., 7., 8.],
[ 9., 10., 11., 12.]],
[[13., 14., 15., 16.],
[17., 18., 19., 20.],
[21., 22., 23., 24.]]]), size=(2, 3, 2, 3), nnz=4,
dtype=torch.float64, layout=torch.sparse_csr)
# _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., 2., 3., 4.],
[ 5., 6., 7., 8.],
[ 9., 10., 11., 12.]],
[[13., 14., 15., 16.],
[17., 18., 19., 20.],
[21., 22., 23., 24.]]], dtype=torch.float64)
########## torch.float32/torch.int64/size=()+(2, 3)+() ##########
# sparse tensor
tensor(crow_indices=tensor([0, 2, 4]),
col_indices=tensor([0, 1, 0, 2]),
values=tensor([1., 2., 3., 4.]), size=(2, 3), nnz=4,
layout=torch.sparse_csr)
# _crow_indices
tensor([0, 2, 4])
# _col_indices
tensor([0, 1, 0, 2])
# _values
tensor([1., 2., 3., 4.])
########## torch.float32/torch.int64/size=()+(0, 0)+() ##########
# sparse tensor
tensor(crow_indices=tensor([0]),
col_indices=tensor([], size=(0,)),
values=tensor([], size=(0,)), size=(0, 0), nnz=0,
layout=torch.sparse_csr)
# _crow_indices
tensor([0])
# _col_indices
tensor([], dtype=torch.int64)
# _values
tensor([])
########## torch.float32/torch.int64/size=(2,)+(2, 3)+() ##########
# 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., 2., 3., 4.],
[5., 6., 7., 8.]]), size=(2, 2, 3), nnz=4,
layout=torch.sparse_csr)
# _crow_indices
tensor([[0, 2, 4],
[0, 3, 4]])
# _col_indices
tensor([[0, 1, 0, 1],
[0, 1, 2, 0]])
# _values
tensor([[1., 2., 3., 4.],
[5., 6., 7., 8.]])
########## torch.float32/torch.int64/size=(2, 3)+(2, 3)+() ##########
# 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., 2., 3., 4.],
[ 5., 6., 7., 8.],
[ 9., 10., 11., 12.]],
[[13., 14., 15., 16.],
[17., 18., 19., 20.],
[21., 22., 23., 24.]]]), size=(2, 3, 2, 3), nnz=4,
layout=torch.sparse_csr)
# _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., 2., 3., 4.],
[ 5., 6., 7., 8.],
[ 9., 10., 11., 12.]],
[[13., 14., 15., 16.],
[17., 18., 19., 20.],
[21., 22., 23., 24.]]])
########## torch.float64/torch.int64/size=()+(2, 3)+() ##########
# sparse tensor
tensor(crow_indices=tensor([0, 2, 4]),
col_indices=tensor([0, 1, 0, 2]),
values=tensor([1., 2., 3., 4.]), size=(2, 3), nnz=4,
dtype=torch.float64, layout=torch.sparse_csr)
# _crow_indices
tensor([0, 2, 4])
# _col_indices
tensor([0, 1, 0, 2])
# _values
tensor([1., 2., 3., 4.], dtype=torch.float64)
########## torch.float64/torch.int64/size=()+(0, 0)+() ##########
# sparse tensor
tensor(crow_indices=tensor([0]),
col_indices=tensor([], size=(0,)),
values=tensor([], size=(0,)), size=(0, 0), nnz=0, dtype=torch.float64,
layout=torch.sparse_csr)
# _crow_indices
tensor([0])
# _col_indices
tensor([], dtype=torch.int64)
# _values
tensor([], dtype=torch.float64)
########## torch.float64/torch.int64/size=(2,)+(2, 3)+() ##########
# 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., 2., 3., 4.],
[5., 6., 7., 8.]]), size=(2, 2, 3), nnz=4,
dtype=torch.float64, layout=torch.sparse_csr)
# _crow_indices
tensor([[0, 2, 4],
[0, 3, 4]])
# _col_indices
tensor([[0, 1, 0, 1],
[0, 1, 2, 0]])
# _values
tensor([[1., 2., 3., 4.],
[5., 6., 7., 8.]], dtype=torch.float64)
########## torch.float64/torch.int64/size=(2, 3)+(2, 3)+() ##########
# 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., 2., 3., 4.],
[ 5., 6., 7., 8.],
[ 9., 10., 11., 12.]],
[[13., 14., 15., 16.],
[17., 18., 19., 20.],
[21., 22., 23., 24.]]]), size=(2, 3, 2, 3), nnz=4,
dtype=torch.float64, layout=torch.sparse_csr)
# _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., 2., 3., 4.],
[ 5., 6., 7., 8.],
[ 9., 10., 11., 12.]],
[[13., 14., 15., 16.],
[17., 18., 19., 20.],
[21., 22., 23., 24.]]], dtype=torch.float64)
########## torch.float32/torch.int32/size=()+(2, 3)+(2,) ##########
# 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=(2, 3, 2), nnz=4,
layout=torch.sparse_csr)
# _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=()+(2, 3)+(4, 2) ##########
# sparse tensor
tensor(crow_indices=tensor([0, 2, 4]),
col_indices=tensor([0, 1, 0, 2]),
values=tensor([[[ 1., 11.],
[ 2., 12.],
[ 3., 13.],
[ 4., 14.]],
[[ 2., 12.],
[ 3., 13.],
[ 4., 14.],
[ 5., 15.]],
[[ 3., 13.],
[ 4., 14.],
[ 5., 15.],
[ 6., 16.]],
[[ 4., 14.],
[ 5., 15.],
[ 6., 16.],
[ 7., 17.]]]), size=(2, 3, 4, 2), nnz=4,
layout=torch.sparse_csr)
# _crow_indices
tensor([0, 2, 4], dtype=torch.int32)
# _col_indices
tensor([0, 1, 0, 2], dtype=torch.int32)
# _values
tensor([[[ 1., 11.],
[ 2., 12.],
[ 3., 13.],
[ 4., 14.]],
[[ 2., 12.],
[ 3., 13.],
[ 4., 14.],
[ 5., 15.]],
[[ 3., 13.],
[ 4., 14.],
[ 5., 15.],
[ 6., 16.]],
[[ 4., 14.],
[ 5., 15.],
[ 6., 16.],
[ 7., 17.]]])
########## torch.float32/torch.int32/size=(2, 3)+(2, 3)+(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.],
[ 2.]],
[[ 2.],
[ 3.]],
[[ 3.],
[ 4.]],
[[ 4.],
[ 5.]]],
[[[ 5.],
[ 6.]],
[[ 6.],
[ 7.]],
[[ 7.],
[ 8.]],
[[ 8.],
[ 9.]]],
[[[ 9.],
[10.]],
[[10.],
[11.]],
[[11.],
[12.]],
[[12.],
[13.]]]],
[[[[13.],
[14.]],
[[14.],
[15.]],
[[15.],
[16.]],
[[16.],
[17.]]],
[[[17.],
[18.]],
[[18.],
[19.]],
[[19.],
[20.]],
[[20.],
[21.]]],
[[[21.],
[22.]],
[[22.],
[23.]],
[[23.],
[24.]],
[[24.],
[25.]]]]]), size=(2, 3, 2, 3, 2, 1), nnz=4,
layout=torch.sparse_csr)
# _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.],
[ 2.]],
[[ 2.],
[ 3.]],
[[ 3.],
[ 4.]],
[[ 4.],
[ 5.]]],
[[[ 5.],
[ 6.]],
[[ 6.],
[ 7.]],
[[ 7.],
[ 8.]],
[[ 8.],
[ 9.]]],
[[[ 9.],
[10.]],
[[10.],
[11.]],
[[11.],
[12.]],
[[12.],
[13.]]]],
[[[[13.],
[14.]],
[[14.],
[15.]],
[[15.],
[16.]],
[[16.],
[17.]]],
[[[17.],
[18.]],
[[18.],
[19.]],
[[19.],
[20.]],
[[20.],
[21.]]],
[[[21.],
[22.]],
[[22.],
[23.]],
[[23.],
[24.]],
[[24.],
[25.]]]]])
########## torch.float64/torch.int32/size=()+(2, 3)+(2,) ##########
# 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=(2, 3, 2), nnz=4, dtype=torch.float64,
layout=torch.sparse_csr)
# _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=()+(2, 3)+(4, 2) ##########
# sparse tensor
tensor(crow_indices=tensor([0, 2, 4]),
col_indices=tensor([0, 1, 0, 2]),
values=tensor([[[ 1., 11.],
[ 2., 12.],
[ 3., 13.],
[ 4., 14.]],
[[ 2., 12.],
[ 3., 13.],
[ 4., 14.],
[ 5., 15.]],
[[ 3., 13.],
[ 4., 14.],
[ 5., 15.],
[ 6., 16.]],
[[ 4., 14.],
[ 5., 15.],
[ 6., 16.],
[ 7., 17.]]]), size=(2, 3, 4, 2), nnz=4,
dtype=torch.float64, layout=torch.sparse_csr)
# _crow_indices
tensor([0, 2, 4], dtype=torch.int32)
# _col_indices
tensor([0, 1, 0, 2], dtype=torch.int32)
# _values
tensor([[[ 1., 11.],
[ 2., 12.],
[ 3., 13.],
[ 4., 14.]],
[[ 2., 12.],
[ 3., 13.],
[ 4., 14.],
[ 5., 15.]],
[[ 3., 13.],
[ 4., 14.],
[ 5., 15.],
[ 6., 16.]],
[[ 4., 14.],
[ 5., 15.],
[ 6., 16.],
[ 7., 17.]]], dtype=torch.float64)
########## torch.float64/torch.int32/size=(2, 3)+(2, 3)+(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.],
[ 2.]],
[[ 2.],
[ 3.]],
[[ 3.],
[ 4.]],
[[ 4.],
[ 5.]]],
[[[ 5.],
[ 6.]],
[[ 6.],
[ 7.]],
[[ 7.],
[ 8.]],
[[ 8.],
[ 9.]]],
[[[ 9.],
[10.]],
[[10.],
[11.]],
[[11.],
[12.]],
[[12.],
[13.]]]],
[[[[13.],
[14.]],
[[14.],
[15.]],
[[15.],
[16.]],
[[16.],
[17.]]],
[[[17.],
[18.]],
[[18.],
[19.]],
[[19.],
[20.]],
[[20.],
[21.]]],
[[[21.],
[22.]],
[[22.],
[23.]],
[[23.],
[24.]],
[[24.],
[25.]]]]]), size=(2, 3, 2, 3, 2, 1), nnz=4,
dtype=torch.float64, layout=torch.sparse_csr)
# _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.],
[ 2.]],
[[ 2.],
[ 3.]],
[[ 3.],
[ 4.]],
[[ 4.],
[ 5.]]],
[[[ 5.],
[ 6.]],
[[ 6.],
[ 7.]],
[[ 7.],
[ 8.]],
[[ 8.],
[ 9.]]],
[[[ 9.],
[10.]],
[[10.],
[11.]],
[[11.],
[12.]],
[[12.],
[13.]]]],
[[[[13.],
[14.]],
[[14.],
[15.]],
[[15.],
[16.]],
[[16.],
[17.]]],
[[[17.],
[18.]],
[[18.],
[19.]],
[[19.],
[20.]],
[[20.],
[21.]]],
[[[21.],
[22.]],
[[22.],
[23.]],
[[23.],
[24.]],
[[24.],
[25.]]]]], dtype=torch.float64)
########## torch.float32/torch.int64/size=()+(2, 3)+(2,) ##########
# 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=(2, 3, 2), nnz=4,
layout=torch.sparse_csr)
# _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=()+(2, 3)+(4, 2) ##########
# sparse tensor
tensor(crow_indices=tensor([0, 2, 4]),
col_indices=tensor([0, 1, 0, 2]),
values=tensor([[[ 1., 11.],
[ 2., 12.],
[ 3., 13.],
[ 4., 14.]],
[[ 2., 12.],
[ 3., 13.],
[ 4., 14.],
[ 5., 15.]],
[[ 3., 13.],
[ 4., 14.],
[ 5., 15.],
[ 6., 16.]],
[[ 4., 14.],
[ 5., 15.],
[ 6., 16.],
[ 7., 17.]]]), size=(2, 3, 4, 2), nnz=4,
layout=torch.sparse_csr)
# _crow_indices
tensor([0, 2, 4])
# _col_indices
tensor([0, 1, 0, 2])
# _values
tensor([[[ 1., 11.],
[ 2., 12.],
[ 3., 13.],
[ 4., 14.]],
[[ 2., 12.],
[ 3., 13.],
[ 4., 14.],
[ 5., 15.]],
[[ 3., 13.],
[ 4., 14.],
[ 5., 15.],
[ 6., 16.]],
[[ 4., 14.],
[ 5., 15.],
[ 6., 16.],
[ 7., 17.]]])
########## torch.float32/torch.int64/size=(2, 3)+(2, 3)+(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.],
[ 2.]],
[[ 2.],
[ 3.]],
[[ 3.],
[ 4.]],
[[ 4.],
[ 5.]]],
[[[ 5.],
[ 6.]],
[[ 6.],
[ 7.]],
[[ 7.],
[ 8.]],
[[ 8.],
[ 9.]]],
[[[ 9.],
[10.]],
[[10.],
[11.]],
[[11.],
[12.]],
[[12.],
[13.]]]],
[[[[13.],
[14.]],
[[14.],
[15.]],
[[15.],
[16.]],
[[16.],
[17.]]],
[[[17.],
[18.]],
[[18.],
[19.]],
[[19.],
[20.]],
[[20.],
[21.]]],
[[[21.],
[22.]],
[[22.],
[23.]],
[[23.],
[24.]],
[[24.],
[25.]]]]]), size=(2, 3, 2, 3, 2, 1), nnz=4,
layout=torch.sparse_csr)
# _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.],
[ 2.]],
[[ 2.],
[ 3.]],
[[ 3.],
[ 4.]],
[[ 4.],
[ 5.]]],
[[[ 5.],
[ 6.]],
[[ 6.],
[ 7.]],
[[ 7.],
[ 8.]],
[[ 8.],
[ 9.]]],
[[[ 9.],
[10.]],
[[10.],
[11.]],
[[11.],
[12.]],
[[12.],
[13.]]]],
[[[[13.],
[14.]],
[[14.],
[15.]],
[[15.],
[16.]],
[[16.],
[17.]]],
[[[17.],
[18.]],
[[18.],
[19.]],
[[19.],
[20.]],
[[20.],
[21.]]],
[[[21.],
[22.]],
[[22.],
[23.]],
[[23.],
[24.]],
[[24.],
[25.]]]]])
########## torch.float64/torch.int64/size=()+(2, 3)+(2,) ##########
# 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=(2, 3, 2), nnz=4, dtype=torch.float64,
layout=torch.sparse_csr)
# _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=()+(2, 3)+(4, 2) ##########
# sparse tensor
tensor(crow_indices=tensor([0, 2, 4]),
col_indices=tensor([0, 1, 0, 2]),
values=tensor([[[ 1., 11.],
[ 2., 12.],
[ 3., 13.],
[ 4., 14.]],
[[ 2., 12.],
[ 3., 13.],
[ 4., 14.],
[ 5., 15.]],
[[ 3., 13.],
[ 4., 14.],
[ 5., 15.],
[ 6., 16.]],
[[ 4., 14.],
[ 5., 15.],
[ 6., 16.],
[ 7., 17.]]]), size=(2, 3, 4, 2), nnz=4,
dtype=torch.float64, layout=torch.sparse_csr)
# _crow_indices
tensor([0, 2, 4])
# _col_indices
tensor([0, 1, 0, 2])
# _values
tensor([[[ 1., 11.],
[ 2., 12.],
[ 3., 13.],
[ 4., 14.]],
[[ 2., 12.],
[ 3., 13.],
[ 4., 14.],
[ 5., 15.]],
[[ 3., 13.],
[ 4., 14.],
[ 5., 15.],
[ 6., 16.]],
[[ 4., 14.],
[ 5., 15.],
[ 6., 16.],
[ 7., 17.]]], dtype=torch.float64)
########## torch.float64/torch.int64/size=(2, 3)+(2, 3)+(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.],
[ 2.]],
[[ 2.],
[ 3.]],
[[ 3.],
[ 4.]],
[[ 4.],
[ 5.]]],
[[[ 5.],
[ 6.]],
[[ 6.],
[ 7.]],
[[ 7.],
[ 8.]],
[[ 8.],
[ 9.]]],
[[[ 9.],
[10.]],
[[10.],
[11.]],
[[11.],
[12.]],
[[12.],
[13.]]]],
[[[[13.],
[14.]],
[[14.],
[15.]],
[[15.],
[16.]],
[[16.],
[17.]]],
[[[17.],
[18.]],
[[18.],
[19.]],
[[19.],
[20.]],
[[20.],
[21.]]],
[[[21.],
[22.]],
[[22.],
[23.]],
[[23.],
[24.]],
[[24.],
[25.]]]]]), size=(2, 3, 2, 3, 2, 1), nnz=4,
dtype=torch.float64, layout=torch.sparse_csr)
# _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.],
[ 2.]],
[[ 2.],
[ 3.]],
[[ 3.],
[ 4.]],
[[ 4.],
[ 5.]]],
[[[ 5.],
[ 6.]],
[[ 6.],
[ 7.]],
[[ 7.],
[ 8.]],
[[ 8.],
[ 9.]]],
[[[ 9.],
[10.]],
[[10.],
[11.]],
[[11.],
[12.]],
[[12.],
[13.]]]],
[[[[13.],
[14.]],
[[14.],
[15.]],
[[15.],
[16.]],
[[16.],
[17.]]],
[[[17.],
[18.]],
[[18.],
[19.]],
[[19.],
[20.]],
[[20.],
[21.]]],
[[[21.],
[22.]],
[[22.],
[23.]],
[[23.],
[24.]],
[[24.],
[25.]]]]], dtype=torch.float64)