amps = tf.reshape(
        tf.cast(params["amps"].get_value(), tf.float64), [params["amps"].shape[0], 1]