def inputs(opts, index):
    value = tf.cast(index, tf.float16)
    return {'inputs': tf.broadcast_to(value, [opts.batch_size, opts.size])}
def inputs(opts, index):
    value = tf.cast(index, tf.float16)
    return {'inputs': tf.broadcast_to(value, [opts.batch_size, opts.size])}