import tensorflow as tf
batch_size = 5
ones = tf.ones([batch_size,20])
logits = tf.layers.dense(ones,10)
print(logits.get_shape())
- 1
- 2
- 3
- 4
- 5
(5,10)
import tensorflow as tf
batch_size = 5
ones = tf.ones([batch_size,8,20])
logits = tf.layers.dense(ones,10)
print(logits.get_shape())
- 1
- 2
- 3
- 4
- 5
(5, 8, 10)
import tensorflow as tf
batch_size = 5
ones = tf.ones([batch_size,6,8,20])
logits = tf.layers.dense(ones,10)
print(logits.get_shape())
- 1
- 2
- 3
- 4
- 5
(5, 6, 8, 10)