コード例 #1
0
ファイル: layers.py プロジェクト: wwxFromTju/reaver-pysc2
 def __init__(self, axis=-1):
     Lambda.__init__(self, lambda x: tf.squeeze(x, axis=axis))
コード例 #2
0
ファイル: layers.py プロジェクト: wwxFromTju/reaver-pysc2
 def __init__(self, size):
     Lambda.__init__(self, lambda x: tf.tile(tf.expand_dims(tf.expand_dims(x, 2), 3), [1, 1, size, size]))
コード例 #3
0
ファイル: layers.py プロジェクト: wwxFromTju/reaver-pysc2
 def __init__(self, scale):
     Lambda.__init__(self, lambda x: tf.cast(x, tf.float32) * scale)
コード例 #4
0
ファイル: layers.py プロジェクト: wwxFromTju/reaver-pysc2
 def __init__(self):
     Lambda.__init__(self, lambda x: tf.log(x + 1e-10))
コード例 #5
0
ファイル: layers.py プロジェクト: wwxFromTju/reaver-pysc2
 def __init__(self, dims):
     Lambda.__init__(self, lambda x: tf.transpose(x, dims))
コード例 #6
0
ファイル: layers.py プロジェクト: wwxFromTju/reaver-pysc2
 def __init__(self, num_splits=2, axis=-1):
     Lambda.__init__(self, lambda x: tf.split(x, num_splits, axis=axis))