在张量流中将无尺寸交换为第二轴

时间:2020-06-04 06:45:28

标签: python tensorflow keras

在Tensorflow中,如果我的模型(用于OCR的CRNN模型)的yPred的形状为[无,否。时间步长类别],如何将其更改为[时间步长,无,无。的课程]?我想将yPred馈送到ctc_beam_search_decoder,它要求形状为[max_time x batch_size x num_classes]。我认为yPred中的None表示批处理大小。 链接到功能: https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/nn/ctc_beam_search_decoder 链接到派生我的模型的网站: https://keras.io/examples/image_ocr/

0 个答案:

没有答案