Wie eine dynamische Form Tensor mit automatischer abgestimmten Form Größe aufgeteilt?

stimmen
0

Angenommen , ich habe einen Platzhalter geformt [None, 512], ich den Tensor auf der ersten dimenstion teilen will dynamisch, mit einer etwa sogar Art und Weise. Zum Beispiel, wenn der Platzhalter mit numpy geformt gespeist wird [5, 512], möchte ich es in zwei Tensoren spalten [2, 512]und [3, 512]. Wenn der Platzhalter mit numpy geformt gespeist wird [7, 512], möchte ich es in zwei Tensoren spalten [3, 512], [4, 512]. Gibt es eine Möglichkeit , das zu tun? Das kann nützlich sein , wenn ich will auf mehr GPUs Daten parallel implementieren , während die Chargengröße dynamisch ist.

Ich habe zu verwenden versucht tf.split( https://www.tensorflow.org/api_docs/python/tf/split ), aber ich habe das zu konstruieren size_tensor.

Veröffentlicht am 07/11/2018 um 23:52
quelle vom benutzer
In anderen Sprachen...                            

Cookies help us deliver our services. By using our services, you agree to our use of cookies. Learn more