@ -413,10 +413,11 @@ class TestVarBase(unittest.TestCase):
var13 = var [ 2 : 10 , 2 : , - 2 : - 1 ]
var13 = var [ 2 : 10 , 2 : , - 2 : - 1 ]
var14 = var [ 1 : - 1 , 0 : 2 , : : - 1 ]
var14 = var [ 1 : - 1 , 0 : 2 , : : - 1 ]
var15 = var [ : : - 1 , : : - 1 , : : - 1 ]
var15 = var [ : : - 1 , : : - 1 , : : - 1 ]
var16 = var [ - 4 : 4 ]
vars = [
vars = [
var , var1 , var2 , var3 , var4 , var5 , var6 , var7 , var8 , var9 , var10 ,
var , var1 , var2 , var3 , var4 , var5 , var6 , var7 , var8 , var9 , var10 ,
var11 , var12 , var13 , var14 , var15
var11 , var12 , var13 , var14 , var15 , var16
]
]
local_out = [ var . numpy ( ) for var in vars ]
local_out = [ var . numpy ( ) for var in vars ]
@ -444,6 +445,7 @@ class TestVarBase(unittest.TestCase):
np . array_equal ( local_out [ 14 ] , tensor_array [ 1 : - 1 , 0 : 2 , : : - 1 ] ) )
np . array_equal ( local_out [ 14 ] , tensor_array [ 1 : - 1 , 0 : 2 , : : - 1 ] ) )
self . assertTrue (
self . assertTrue (
np . array_equal ( local_out [ 15 ] , tensor_array [ : : - 1 , : : - 1 , : : - 1 ] ) )
np . array_equal ( local_out [ 15 ] , tensor_array [ : : - 1 , : : - 1 , : : - 1 ] ) )
self . assertTrue ( np . array_equal ( local_out [ 16 ] , tensor_array [ - 4 : 4 ] ) )
def _test_for_var ( self ) :
def _test_for_var ( self ) :
np_value = np . random . random ( ( 30 , 100 , 100 ) ) . astype ( ' float32 ' )
np_value = np . random . random ( ( 30 , 100 , 100 ) ) . astype ( ' float32 ' )
@ -464,6 +466,9 @@ class TestVarBase(unittest.TestCase):
with self . assertRaises ( IndexError ) :
with self . assertRaises ( IndexError ) :
y = var [ self . shape [ 0 ] ]
y = var [ self . shape [ 0 ] ]
with self . assertRaises ( IndexError ) :
y = var [ 0 - self . shape [ 0 ] - 1 ]
def test_var_base_to_np ( self ) :
def test_var_base_to_np ( self ) :
with fluid . dygraph . guard ( ) :
with fluid . dygraph . guard ( ) :
var = fluid . dygraph . to_variable ( self . array )
var = fluid . dygraph . to_variable ( self . array )