saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/test/expect/TestSparseCompressedCUDA.test_print_SparseCSC_cuda.expect

1411 lines
34 KiB
Text
Raw Blame History

########## torch.float32/torch.int32/size=()+(3, 2)+() ##########
# sparse tensor
tensor(ccol_indices=tensor([0, 2, 4]),
row_indices=tensor([0, 1, 0, 2]),
values=tensor([1., 2., 3., 4.]), device='cuda:0', size=(3, 2), nnz=4,
layout=torch.sparse_csc)
# _ccol_indices
tensor([0, 2, 4], device='cuda:0', dtype=torch.int32)
# _row_indices
tensor([0, 1, 0, 2], device='cuda:0', dtype=torch.int32)
# _values
tensor([1., 2., 3., 4.], device='cuda:0')
########## torch.float32/torch.int32/size=()+(0, 0)+() ##########
# sparse tensor
tensor(ccol_indices=tensor([0]),
row_indices=tensor([], size=(0,)),
values=tensor([], size=(0,)), device='cuda:0', size=(0, 0), nnz=0,
layout=torch.sparse_csc)
# _ccol_indices
tensor([0], device='cuda:0', dtype=torch.int32)
# _row_indices
tensor([], device='cuda:0', dtype=torch.int32)
# _values
tensor([], device='cuda:0')
########## torch.float32/torch.int32/size=(2,)+(3, 2)+() ##########
# sparse tensor
tensor(ccol_indices=tensor([[0, 2, 4],
[0, 3, 4]]),
row_indices=tensor([[0, 1, 0, 1],
[0, 1, 2, 0]]),
values=tensor([[1., 2., 3., 4.],
[5., 6., 7., 8.]]), device='cuda:0', size=(2, 3, 2),
nnz=4, layout=torch.sparse_csc)
# _ccol_indices
tensor([[0, 2, 4],
[0, 3, 4]], device='cuda:0', dtype=torch.int32)
# _row_indices
tensor([[0, 1, 0, 1],
[0, 1, 2, 0]], device='cuda:0', dtype=torch.int32)
# _values
tensor([[1., 2., 3., 4.],
[5., 6., 7., 8.]], device='cuda:0')
########## torch.float32/torch.int32/size=(2, 3)+(3, 2)+() ##########
# sparse tensor
tensor(ccol_indices=tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]]),
row_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.]]]), device='cuda:0',
size=(2, 3, 3, 2), nnz=4, layout=torch.sparse_csc)
# _ccol_indices
tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]], device='cuda:0', dtype=torch.int32)
# _row_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]]], device='cuda:0', 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.]]], device='cuda:0')
########## torch.float64/torch.int32/size=()+(3, 2)+() ##########
# sparse tensor
tensor(ccol_indices=tensor([0, 2, 4]),
row_indices=tensor([0, 1, 0, 2]),
values=tensor([1., 2., 3., 4.]), device='cuda:0', size=(3, 2), nnz=4,
dtype=torch.float64, layout=torch.sparse_csc)
# _ccol_indices
tensor([0, 2, 4], device='cuda:0', dtype=torch.int32)
# _row_indices
tensor([0, 1, 0, 2], device='cuda:0', dtype=torch.int32)
# _values
tensor([1., 2., 3., 4.], device='cuda:0', dtype=torch.float64)
########## torch.float64/torch.int32/size=()+(0, 0)+() ##########
# sparse tensor
tensor(ccol_indices=tensor([0]),
row_indices=tensor([], size=(0,)),
values=tensor([], size=(0,)), device='cuda:0', size=(0, 0), nnz=0,
dtype=torch.float64, layout=torch.sparse_csc)
# _ccol_indices
tensor([0], device='cuda:0', dtype=torch.int32)
# _row_indices
tensor([], device='cuda:0', dtype=torch.int32)
# _values
tensor([], device='cuda:0', dtype=torch.float64)
########## torch.float64/torch.int32/size=(2,)+(3, 2)+() ##########
# sparse tensor
tensor(ccol_indices=tensor([[0, 2, 4],
[0, 3, 4]]),
row_indices=tensor([[0, 1, 0, 1],
[0, 1, 2, 0]]),
values=tensor([[1., 2., 3., 4.],
[5., 6., 7., 8.]]), device='cuda:0', size=(2, 3, 2),
nnz=4, dtype=torch.float64, layout=torch.sparse_csc)
# _ccol_indices
tensor([[0, 2, 4],
[0, 3, 4]], device='cuda:0', dtype=torch.int32)
# _row_indices
tensor([[0, 1, 0, 1],
[0, 1, 2, 0]], device='cuda:0', dtype=torch.int32)
# _values
tensor([[1., 2., 3., 4.],
[5., 6., 7., 8.]], device='cuda:0', dtype=torch.float64)
########## torch.float64/torch.int32/size=(2, 3)+(3, 2)+() ##########
# sparse tensor
tensor(ccol_indices=tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]]),
row_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.]]]), device='cuda:0',
size=(2, 3, 3, 2), nnz=4, dtype=torch.float64, layout=torch.sparse_csc)
# _ccol_indices
tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]], device='cuda:0', dtype=torch.int32)
# _row_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]]], device='cuda:0', 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.]]], device='cuda:0', dtype=torch.float64)
########## torch.float32/torch.int64/size=()+(3, 2)+() ##########
# sparse tensor
tensor(ccol_indices=tensor([0, 2, 4]),
row_indices=tensor([0, 1, 0, 2]),
values=tensor([1., 2., 3., 4.]), device='cuda:0', size=(3, 2), nnz=4,
layout=torch.sparse_csc)
# _ccol_indices
tensor([0, 2, 4], device='cuda:0')
# _row_indices
tensor([0, 1, 0, 2], device='cuda:0')
# _values
tensor([1., 2., 3., 4.], device='cuda:0')
########## torch.float32/torch.int64/size=()+(0, 0)+() ##########
# sparse tensor
tensor(ccol_indices=tensor([0]),
row_indices=tensor([], size=(0,)),
values=tensor([], size=(0,)), device='cuda:0', size=(0, 0), nnz=0,
layout=torch.sparse_csc)
# _ccol_indices
tensor([0], device='cuda:0')
# _row_indices
tensor([], device='cuda:0', dtype=torch.int64)
# _values
tensor([], device='cuda:0')
########## torch.float32/torch.int64/size=(2,)+(3, 2)+() ##########
# sparse tensor
tensor(ccol_indices=tensor([[0, 2, 4],
[0, 3, 4]]),
row_indices=tensor([[0, 1, 0, 1],
[0, 1, 2, 0]]),
values=tensor([[1., 2., 3., 4.],
[5., 6., 7., 8.]]), device='cuda:0', size=(2, 3, 2),
nnz=4, layout=torch.sparse_csc)
# _ccol_indices
tensor([[0, 2, 4],
[0, 3, 4]], device='cuda:0')
# _row_indices
tensor([[0, 1, 0, 1],
[0, 1, 2, 0]], device='cuda:0')
# _values
tensor([[1., 2., 3., 4.],
[5., 6., 7., 8.]], device='cuda:0')
########## torch.float32/torch.int64/size=(2, 3)+(3, 2)+() ##########
# sparse tensor
tensor(ccol_indices=tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]]),
row_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.]]]), device='cuda:0',
size=(2, 3, 3, 2), nnz=4, layout=torch.sparse_csc)
# _ccol_indices
tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]], device='cuda:0')
# _row_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]]], device='cuda:0')
# _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.]]], device='cuda:0')
########## torch.float64/torch.int64/size=()+(3, 2)+() ##########
# sparse tensor
tensor(ccol_indices=tensor([0, 2, 4]),
row_indices=tensor([0, 1, 0, 2]),
values=tensor([1., 2., 3., 4.]), device='cuda:0', size=(3, 2), nnz=4,
dtype=torch.float64, layout=torch.sparse_csc)
# _ccol_indices
tensor([0, 2, 4], device='cuda:0')
# _row_indices
tensor([0, 1, 0, 2], device='cuda:0')
# _values
tensor([1., 2., 3., 4.], device='cuda:0', dtype=torch.float64)
########## torch.float64/torch.int64/size=()+(0, 0)+() ##########
# sparse tensor
tensor(ccol_indices=tensor([0]),
row_indices=tensor([], size=(0,)),
values=tensor([], size=(0,)), device='cuda:0', size=(0, 0), nnz=0,
dtype=torch.float64, layout=torch.sparse_csc)
# _ccol_indices
tensor([0], device='cuda:0')
# _row_indices
tensor([], device='cuda:0', dtype=torch.int64)
# _values
tensor([], device='cuda:0', dtype=torch.float64)
########## torch.float64/torch.int64/size=(2,)+(3, 2)+() ##########
# sparse tensor
tensor(ccol_indices=tensor([[0, 2, 4],
[0, 3, 4]]),
row_indices=tensor([[0, 1, 0, 1],
[0, 1, 2, 0]]),
values=tensor([[1., 2., 3., 4.],
[5., 6., 7., 8.]]), device='cuda:0', size=(2, 3, 2),
nnz=4, dtype=torch.float64, layout=torch.sparse_csc)
# _ccol_indices
tensor([[0, 2, 4],
[0, 3, 4]], device='cuda:0')
# _row_indices
tensor([[0, 1, 0, 1],
[0, 1, 2, 0]], device='cuda:0')
# _values
tensor([[1., 2., 3., 4.],
[5., 6., 7., 8.]], device='cuda:0', dtype=torch.float64)
########## torch.float64/torch.int64/size=(2, 3)+(3, 2)+() ##########
# sparse tensor
tensor(ccol_indices=tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]]),
row_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.]]]), device='cuda:0',
size=(2, 3, 3, 2), nnz=4, dtype=torch.float64, layout=torch.sparse_csc)
# _ccol_indices
tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]], device='cuda:0')
# _row_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]]], device='cuda:0')
# _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.]]], device='cuda:0', dtype=torch.float64)
########## torch.float32/torch.int32/size=()+(3, 2)+(2,) ##########
# sparse tensor
tensor(ccol_indices=tensor([0, 2, 4]),
row_indices=tensor([0, 1, 0, 2]),
values=tensor([[1., 2.],
[2., 3.],
[3., 4.],
[4., 5.]]), device='cuda:0', size=(3, 2, 2), nnz=4,
layout=torch.sparse_csc)
# _ccol_indices
tensor([0, 2, 4], device='cuda:0', dtype=torch.int32)
# _row_indices
tensor([0, 1, 0, 2], device='cuda:0', dtype=torch.int32)
# _values
tensor([[1., 2.],
[2., 3.],
[3., 4.],
[4., 5.]], device='cuda:0')
########## torch.float32/torch.int32/size=()+(3, 2)+(4, 2) ##########
# sparse tensor
tensor(ccol_indices=tensor([0, 2, 4]),
row_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.]]]), device='cuda:0', size=(3, 2, 4, 2),
nnz=4, layout=torch.sparse_csc)
# _ccol_indices
tensor([0, 2, 4], device='cuda:0', dtype=torch.int32)
# _row_indices
tensor([0, 1, 0, 2], device='cuda:0', 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.]]], device='cuda:0')
########## torch.float32/torch.int32/size=(2, 3)+(3, 2)+(2, 1) ##########
# sparse tensor
tensor(ccol_indices=tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]]),
row_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.]]]]]), device='cuda:0', size=(2, 3, 3, 2, 2, 1),
nnz=4, layout=torch.sparse_csc)
# _ccol_indices
tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]], device='cuda:0', dtype=torch.int32)
# _row_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]]], device='cuda:0', 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.]]]]], device='cuda:0')
########## torch.float64/torch.int32/size=()+(3, 2)+(2,) ##########
# sparse tensor
tensor(ccol_indices=tensor([0, 2, 4]),
row_indices=tensor([0, 1, 0, 2]),
values=tensor([[1., 2.],
[2., 3.],
[3., 4.],
[4., 5.]]), device='cuda:0', size=(3, 2, 2), nnz=4,
dtype=torch.float64, layout=torch.sparse_csc)
# _ccol_indices
tensor([0, 2, 4], device='cuda:0', dtype=torch.int32)
# _row_indices
tensor([0, 1, 0, 2], device='cuda:0', dtype=torch.int32)
# _values
tensor([[1., 2.],
[2., 3.],
[3., 4.],
[4., 5.]], device='cuda:0', dtype=torch.float64)
########## torch.float64/torch.int32/size=()+(3, 2)+(4, 2) ##########
# sparse tensor
tensor(ccol_indices=tensor([0, 2, 4]),
row_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.]]]), device='cuda:0', size=(3, 2, 4, 2),
nnz=4, dtype=torch.float64, layout=torch.sparse_csc)
# _ccol_indices
tensor([0, 2, 4], device='cuda:0', dtype=torch.int32)
# _row_indices
tensor([0, 1, 0, 2], device='cuda:0', 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.]]], device='cuda:0', dtype=torch.float64)
########## torch.float64/torch.int32/size=(2, 3)+(3, 2)+(2, 1) ##########
# sparse tensor
tensor(ccol_indices=tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]]),
row_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.]]]]]), device='cuda:0', size=(2, 3, 3, 2, 2, 1),
nnz=4, dtype=torch.float64, layout=torch.sparse_csc)
# _ccol_indices
tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]], device='cuda:0', dtype=torch.int32)
# _row_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]]], device='cuda:0', 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.]]]]], device='cuda:0', dtype=torch.float64)
########## torch.float32/torch.int64/size=()+(3, 2)+(2,) ##########
# sparse tensor
tensor(ccol_indices=tensor([0, 2, 4]),
row_indices=tensor([0, 1, 0, 2]),
values=tensor([[1., 2.],
[2., 3.],
[3., 4.],
[4., 5.]]), device='cuda:0', size=(3, 2, 2), nnz=4,
layout=torch.sparse_csc)
# _ccol_indices
tensor([0, 2, 4], device='cuda:0')
# _row_indices
tensor([0, 1, 0, 2], device='cuda:0')
# _values
tensor([[1., 2.],
[2., 3.],
[3., 4.],
[4., 5.]], device='cuda:0')
########## torch.float32/torch.int64/size=()+(3, 2)+(4, 2) ##########
# sparse tensor
tensor(ccol_indices=tensor([0, 2, 4]),
row_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.]]]), device='cuda:0', size=(3, 2, 4, 2),
nnz=4, layout=torch.sparse_csc)
# _ccol_indices
tensor([0, 2, 4], device='cuda:0')
# _row_indices
tensor([0, 1, 0, 2], device='cuda:0')
# _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.]]], device='cuda:0')
########## torch.float32/torch.int64/size=(2, 3)+(3, 2)+(2, 1) ##########
# sparse tensor
tensor(ccol_indices=tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]]),
row_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.]]]]]), device='cuda:0', size=(2, 3, 3, 2, 2, 1),
nnz=4, layout=torch.sparse_csc)
# _ccol_indices
tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]], device='cuda:0')
# _row_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]]], device='cuda:0')
# _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.]]]]], device='cuda:0')
########## torch.float64/torch.int64/size=()+(3, 2)+(2,) ##########
# sparse tensor
tensor(ccol_indices=tensor([0, 2, 4]),
row_indices=tensor([0, 1, 0, 2]),
values=tensor([[1., 2.],
[2., 3.],
[3., 4.],
[4., 5.]]), device='cuda:0', size=(3, 2, 2), nnz=4,
dtype=torch.float64, layout=torch.sparse_csc)
# _ccol_indices
tensor([0, 2, 4], device='cuda:0')
# _row_indices
tensor([0, 1, 0, 2], device='cuda:0')
# _values
tensor([[1., 2.],
[2., 3.],
[3., 4.],
[4., 5.]], device='cuda:0', dtype=torch.float64)
########## torch.float64/torch.int64/size=()+(3, 2)+(4, 2) ##########
# sparse tensor
tensor(ccol_indices=tensor([0, 2, 4]),
row_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.]]]), device='cuda:0', size=(3, 2, 4, 2),
nnz=4, dtype=torch.float64, layout=torch.sparse_csc)
# _ccol_indices
tensor([0, 2, 4], device='cuda:0')
# _row_indices
tensor([0, 1, 0, 2], device='cuda:0')
# _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.]]], device='cuda:0', dtype=torch.float64)
########## torch.float64/torch.int64/size=(2, 3)+(3, 2)+(2, 1) ##########
# sparse tensor
tensor(ccol_indices=tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]]),
row_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.]]]]]), device='cuda:0', size=(2, 3, 3, 2, 2, 1),
nnz=4, dtype=torch.float64, layout=torch.sparse_csc)
# _ccol_indices
tensor([[[0, 2, 4],
[0, 3, 4],
[0, 1, 4]],
[[0, 1, 4],
[0, 2, 4],
[0, 3, 4]]], device='cuda:0')
# _row_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]]], device='cuda:0')
# _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.]]]]], device='cuda:0', dtype=torch.float64)
Reference in a new issue View git blame Copy permalink