test_layers_extend.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import os
import unittest

import tensorflow as tf

import tensorlayer as tl
from tests.utils import CustomTestCase

os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'


class Layer_Extend_Test(CustomTestCase):

    @classmethod
    def setUpClass(cls):
        pass

    @classmethod
    def tearDownClass(cls):
        pass

    def test_expand_dims(self):
        x = tl.layers.Input([8, 3])
        expandlayer = tl.layers.ExpandDims(axis=-1)
        y = expandlayer(x)
        print(expandlayer)
        self.assertEqual(y.get_shape().as_list(), [8, 3, 1])

    def test_tile(self):
        x = tl.layers.Input([8, 3])
        tilelayer = tl.layers.Tile(multiples=[2, 3])
        y = tilelayer(x)
        print(tilelayer)
        self.assertEqual(y.get_shape().as_list(), [16, 9])


if __name__ == '__main__':

    unittest.main()