Jak rozdělit dynamicky tvarovaný tensor, s automatickým naladěné velikosti tvar?

hlasů
0

Předpokládejme, že mám zástupný tvaru [None, 512], chci rozdělit tensor na první dimenstion dynamicky, s přibližně vyrovnat způsobem. Například, pokud je zástupný symbol je napájen NumPy tvarovanými [5, 512], chci ji rozdělit do dvou tenzorů [2, 512]a [3, 512]. Je-li zástupný krmena NumPy tvarovanými [7, 512], chci ji rozdělit do dvou tenzorů [3, 512], [4, 512]. Existuje nějaký způsob, jak to udělat? Tyto události mohou být užitečné, pokud chci realizovat dat paralelně na více GPU, zatímco velikost dávky je dynamická.

Snažil jsem se použít tf.split( https://www.tensorflow.org/api_docs/python/tf/split ), ale nepodařilo se mi konstruovat size_tensor.

Položena 07/11/2018 v 23:52
zdroj uživatelem
V jiných jazycích...                            

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