|
|
|
@ -51,8 +51,8 @@ class TestExpandAsOpRank1(OpTest):
|
|
|
|
|
class TestExpandAsOpRank2(OpTest):
|
|
|
|
|
def setUp(self):
|
|
|
|
|
self.op_type = "expand_as"
|
|
|
|
|
x = np.random.rand(2, 3).astype("float64")
|
|
|
|
|
target_tensor = np.random.rand(4, 6).astype("float64")
|
|
|
|
|
x = np.random.rand(10, 12).astype("float64")
|
|
|
|
|
target_tensor = np.random.rand(20, 24).astype("float64")
|
|
|
|
|
self.inputs = {'X': x, 'target_tensor': target_tensor}
|
|
|
|
|
self.attrs = {}
|
|
|
|
|
bcast_dims = bcast(x, target_tensor)
|
|
|
|
@ -69,8 +69,8 @@ class TestExpandAsOpRank2(OpTest):
|
|
|
|
|
class TestExpandAsOpRank3(OpTest):
|
|
|
|
|
def setUp(self):
|
|
|
|
|
self.op_type = "expand_as"
|
|
|
|
|
x = np.random.rand(2, 3, 3).astype("float64")
|
|
|
|
|
target_tensor = np.random.rand(4, 6, 6).astype("float64")
|
|
|
|
|
x = np.random.rand(2, 3, 20).astype("float64")
|
|
|
|
|
target_tensor = np.random.rand(4, 6, 40).astype("float64")
|
|
|
|
|
self.inputs = {'X': x, 'target_tensor': target_tensor}
|
|
|
|
|
self.attrs = {}
|
|
|
|
|
bcast_dims = bcast(x, target_tensor)
|
|
|
|
@ -87,8 +87,8 @@ class TestExpandAsOpRank3(OpTest):
|
|
|
|
|
class TestExpandAsOpRank4(OpTest):
|
|
|
|
|
def setUp(self):
|
|
|
|
|
self.op_type = "expand_as"
|
|
|
|
|
x = np.random.rand(1, 1, 3, 16).astype("float64")
|
|
|
|
|
target_tensor = np.random.rand(4, 6, 6, 32).astype("float64")
|
|
|
|
|
x = np.random.rand(1, 1, 7, 16).astype("float64")
|
|
|
|
|
target_tensor = np.random.rand(4, 6, 14, 32).astype("float64")
|
|
|
|
|
self.inputs = {'X': x, 'target_tensor': target_tensor}
|
|
|
|
|
self.attrs = {}
|
|
|
|
|
bcast_dims = bcast(x, target_tensor)
|
|
|
|
|