tf.layers.dense

import tensorflow as tf
batch_size = 5
ones = tf.ones([batch_size,20])
logits = tf.layers.dense(ones,10)
print(logits.get_shape())

import tensorflow as tf
batch_size = 5
ones = tf.ones([batch_size,8,20])
logits = tf.layers.dense(ones,10)
print(logits.get_shape())

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())

‘‘‘
(5, 10)
(5, 8, 10)
(5, 6, 8, 10)
‘‘‘