def testDynamicTileAndConcat(self):
# image = (1 X 4 X 4 X 1)
image = [[1, 2, 3, 4],
[2, 4, 5, 6],
[7, 8, 9, 10],
[7, 9, 10, 1]]
image_t = tf.expand_dims(tf.expand_dims(image, axis=0), axis=-1)
image_t = tf.cast(image_t, dtype=tf.float32)
# latent = (1 X 2)
latent = np.array([[90, 100]])
latent_t = tf.cast(tf.convert_to_tensor(latent), dtype=tf.float32)
tiled = common_video.tile_and_concat(
image_t, latent_t)
tiled_np, image_np = self.evaluate([tiled, image_t])
tiled_latent = tiled_np[0, :, :, -1]
self.assertAllEqual(tiled_np.shape, (1, 4, 4, 2))
self.assertAllEqual(tiled_np[:, :, :, :1], image_np)
self.assertAllEqual(
tiled_latent,
[[90, 90, 90, 90],
[100, 100, 100, 100],
[90, 90, 90, 90],
[100, 100, 100, 100]])
def testDynamicTileAndConcat(self):
# image = (1 X 4 X 4 X 1)
image = [[1, 2, 3, 4],
[2, 4, 5, 6],
[7, 8, 9, 10],
[7, 9, 10, 1]]
image_t = tf.expand_dims(tf.expand_dims(image, axis=0), axis=-1)
image_t = tf.cast(image_t, dtype=tf.float32)
# latent = (1 X 2)
latent = np.array([[90, 100]])
latent_t = tf.cast(tf.convert_to_tensor(latent), dtype=tf.float32)
tiled = common_video.tile_and_concat(
image_t, latent_t)
tiled_np, image_np = self.evaluate([tiled, image_t])
tiled_latent = tiled_np[0, :, :, -1]
self.assertAllEqual(tiled_np.shape, (1, 4, 4, 2))
self.assertAllEqual(tiled_np[:, :, :, :1], image_np)
self.assertAllEqual(
tiled_latent,
[[90, 90, 90, 90],
[100, 100, 100, 100],
[90, 90, 90, 90],
[100, 100, 100, 100]])
