σ VΎχXc@`s»dZddlmZddlmZddlmZddlmZddlmZddlm Z d„Z d „Z d „Z dd „Zddd „Zdd „Zddd„ZdS(s™A small library of functions dealing with LSTMs applied to images. Tensors in this library generally have the shape (num_images, height, width, depth). i(tabsolute_import(tdivision(tprint_function(tlstm1d(t array_ops(tvariable_scopecC`s|jƒjƒS(s)Get the shape of a tensor as an int list.(t get_shapetas_list(ttensor((si/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/ndlstm/python/lstm2d.pyt_shapescC`sSt|ƒ\}}}}tj|ddddgƒ}tj|||||gƒS(sΊConvert a batch of images into a batch of sequences. Args: tensor: a (num_images, height, width, depth) tensor Returns: (width, num_images*height, depth) sequence tensor iiii(R Rt transposetreshape(Rtnum_image_batchestheighttwidthtdeptht transposed((si/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/ndlstm/python/lstm2d.pytimages_to_sequence"s  cC`sYt|ƒ\}}}||}tj|||||gƒ}tj|ddddgƒS(sιConvert a batch of sequences into a batch of images. Args: tensor: (num_steps, num_batches, depth) sequence tensor num_image_batches: the number of image batches Returns: (num_images, height, width, depth) tensor iiii(R RR R (RR Rt num_batchesRR treshaped((si/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/ndlstm/python/lstm2d.pytsequence_to_images2s   c C`sΤtj|d|gƒΆt|ƒ\}}}}t|ƒ}tjdƒtj||dƒ}WdQXtjdƒ%tj|||dddƒ}WdQXtj||gdƒ}t||ƒ} | SWdQXdS(sSRun an LSTM bidirectionally over all the rows of each image. Args: images: (num_images, height, width, depth) tensor num_filters_out: output depth scope: optional scope name Returns: (num_images, height, width, num_filters_out) tensor, where num_steps is width and new num_batches is num_image_batches * height tHorizontalLstmtlriNtrltreversei(RR RRt ndlstm_baseRtconcatR( timagestnum_filters_outtscopet batch_sizet_tsequencethidden_sequence_lrthidden_sequence_rltoutput_sequencetoutput((si/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/ndlstm/python/lstm2d.pythorizontal_lstmDs    cC`s¬tj|d|gƒŽ|dkr.|}nt||ƒ}tjdƒ2tj|ddddgƒ}t||ƒ}WdQXtj|ddddgƒ}|SWdQXdS(s(Run bidirectional LSTMs first horizontally then vertically. Args: images: (num_images, height, width, depth) tensor num_filters_out: output layer depth nhidden: hidden layer depth scope: optional scope name Returns: (num_images, height, width, num_filters_out) tensor t SeparableLstmtverticaliiiiN(RtNoneR%RR (RRtnhiddenRthiddenRtoutput_transposedR$((si/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/ndlstm/python/lstm2d.pytseparable_lstm^s   c C`s©tj|d|gƒ‹t|ƒ\}}}}tj|ddddgƒ}tj|||||gƒ}tj||ƒ} tj| |||gƒ} | SWdQXdS(sReduce an image to a sequence by scanning an LSTM vertically. Args: images: (num_images, height, width, depth) tensor num_filters_out: output layer depth scope: optional scope name Returns: A (width, num_images, num_filters_out) sequence. tReduceToSequenceiiiiN(RR RR R Rtsequence_to_final( RRRRR RRRRtreducedR$((si/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/ndlstm/python/lstm2d.pytreduce_to_sequenceus  cC`stj|d|gƒπ|p"|}t|ƒ\}}}}tj|ddddgƒ}tj|||||gƒ} tjdƒMtj| |ƒ} tj| |||gƒ} tj| dddgƒ} WdQXtjdƒtj| |ƒ} WdQX| SWdQXdS( s4Reduce an image to a final state by running two LSTMs. Args: images: (num_images, height, width, depth) tensor num_filters_out: output layer depth nhidden: hidden layer depth (defaults to num_filters_out) scope: optional scope name Returns: A (num_images, num_filters_out) batch. t ReduceToFinaliiiitreduce1Ntreduce2(RR RR R RR.(RRR)RRR RRRRR/ttransposed_hiddenR*R$((si/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/ndlstm/python/lstm2d.pytreduce_to_finalŠs    !N(t__doc__t __future__RRRt tensorflow.contrib.ndlstm.pythonRttensorflow.python.opsRRR RRR(R%R,R0R5(((si/tmp/pip-build-h1VYrz/tensorflow/tensorflow-1.0.1.data/purelib/tensorflow/contrib/ndlstm/python/lstm2d.pyts