File tree Expand file tree Collapse file tree 2 files changed +6
-8
lines changed
python/paddle/fluid/tests/unittests Expand file tree Collapse file tree 2 files changed +6
-8
lines changed Original file line number Diff line number Diff line change @@ -40,12 +40,11 @@ def test_tensor_fill_true(self):
40
40
for dtype in typelist :
41
41
var = 1.
42
42
tensor = paddle .to_tensor (np_arr , place = p , dtype = dtype )
43
- newtensor = tensor .clone ()
44
- newtensor [...] = var
43
+ target = tensor .numpy ()
44
+ target [...] = var
45
45
46
46
tensor .fill_ (var ) #var type is basic type in typelist
47
- self .assertEqual ((tensor .numpy () == newtensor .numpy ()).all (),
48
- True )
47
+ self .assertEqual ((tensor .numpy () == target ).all (), True )
49
48
50
49
def test_tensor_fill_backward (self ):
51
50
typelist = ['float32' ]
Original file line number Diff line number Diff line change @@ -35,12 +35,11 @@ def test_tensor_fill_true(self):
35
35
np .array (six .moves .range (np .prod (self .shape ))), self .shape )
36
36
for dtype in typelist :
37
37
tensor = paddle .to_tensor (np_arr , place = p , dtype = dtype )
38
- newtensor = tensor .clone ()
39
- newtensor [...] = 0
38
+ target = tensor .numpy ()
39
+ target [...] = 0
40
40
41
41
tensor .zero_ ()
42
- self .assertEqual (
43
- (tensor .numpy () == newtensor .numpy ()).all ().item (), True )
42
+ self .assertEqual ((tensor .numpy () == target ).all ().item (), True )
44
43
45
44
46
45
if __name__ == '__main__' :
You can’t perform that action at this time.
0 commit comments