new_shorter_edge = tf.constant(400)
new_height, new_width = tf.cond(height_smaller_than_width, lambda :
    new_shorter_edge, width / height * new_shorter_edge, lambda :
    new_shorter_edge, height / width * new_shorter_edge)
TensorFlow Изменение размера тензора изображения в динамическую форму