Commit 76cf35de authored by Neal Wu's avatar Neal Wu Committed by GitHub
Browse files

Add named arguments to tf.concat and use conv2d instead of convolution2d

parent 514a10de
......@@ -128,13 +128,13 @@ class AdversarialCrypto(object):
"""
if key is not None:
combined_message = tf.concat([message, key], 1)
combined_message = tf.concat(axis=1, values=[message, key])
else:
combined_message = message
# Ensure that all variables created are in the specified collection.
with tf.contrib.framework.arg_scope(
[tf.contrib.layers.fully_connected, tf.contrib.layers.convolution2d],
[tf.contrib.layers.fully_connected, tf.contrib.layers.conv2d],
variables_collections=[collection]):
fc = tf.contrib.layers.fully_connected(
......@@ -147,13 +147,13 @@ class AdversarialCrypto(object):
# and then squeezing it back down).
fc = tf.expand_dims(fc, 2)
# 2,1 -> 1,2
conv = tf.contrib.layers.convolution2d(
conv = tf.contrib.layers.conv2d(
fc, 2, 2, 2, 'SAME', activation_fn=tf.nn.sigmoid)
# 1,2 -> 1, 2
conv = tf.contrib.layers.convolution2d(
conv = tf.contrib.layers.conv2d(
conv, 2, 1, 1, 'SAME', activation_fn=tf.nn.sigmoid)
# 1,2 -> 1, 1
conv = tf.contrib.layers.convolution2d(
conv = tf.contrib.layers.conv2d(
conv, 1, 1, 1, 'SAME', activation_fn=tf.nn.tanh)
conv = tf.squeeze(conv, 2)
return conv
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment