|
|
@ -57,7 +57,8 @@ class TestSimilarityFocusOp(OpTest):
|
|
|
|
if cnt == min(y_dim, z_dim):
|
|
|
|
if cnt == min(y_dim, z_dim):
|
|
|
|
break
|
|
|
|
break
|
|
|
|
channel[index] = -1
|
|
|
|
channel[index] = -1
|
|
|
|
res = res.reshape(1, y_dim, z_dim).repeat([x_dim], axis=0)
|
|
|
|
res = res.reshape(1, y_dim, z_dim)
|
|
|
|
|
|
|
|
res = res.repeat([x_dim], axis=0)
|
|
|
|
res = res.reshape(1, x_dim, y_dim, z_dim)
|
|
|
|
res = res.reshape(1, x_dim, y_dim, z_dim)
|
|
|
|
if output is not None:
|
|
|
|
if output is not None:
|
|
|
|
output = np.concatenate((output, res), axis=0)
|
|
|
|
output = np.concatenate((output, res), axis=0)
|
|
|
|