fix examples

PaddlePaddle · Apr 14, 2022 · 0768215 · 0768215 · paddle-bot-old · Apr 15, 2022
1 parent 83a12d6
commit 0768215
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/paddle/sparse/layer/conv.py b/python/paddle/sparse/layer/conv.py
@@ -210,6 +210,8 @@ class Conv3D(_Conv3D):
           with _test_eager_guard():
             indices = [[0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 1, 2], [1, 3, 2, 3]]
             values = [[1], [2], [3], [4]]
+            indices = paddle.to_tensor(indices, dtype='int32')
+            values = paddle.to_tensor(values, dtype='float32')
             dense_shape = [1, 1, 3, 4, 1]
             sparse_x = paddle.sparse.sparse_coo_tensor(indices, values, dense_shape, stop_gradient=True) 
             conv = paddle.sparse.Conv3D(1, 1, (1, 3, 3))
@@ -342,6 +344,8 @@ class SubmConv3D(_Conv3D):
             indices = [[0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 1, 2], [1, 3, 2, 3]]
             values = [[1], [2], [3], [4]]
             dense_shape = [1, 1, 3, 4, 1]
+            indices = paddle.to_tensor(indices, dtype='int32')
+            values = paddle.to_tensor(values, dtype='float32')
             sparse_x = paddle.sparse.sparse_coo_tensor(indices, values, dense_shape, stop_gradient=True) 
             subm_conv = paddle.sparse.SubmConv3D(1, 1, (1, 3, 3))
             y = subm_conv(sparse_x)