Πώς να χωρίσετε ένα δυναμικό σχήμα τανυστής, με αυτόματη συντονισμένοι μέγεθος το σχήμα;

ψήφοι
0

Ας υποθέσουμε ότι έχω ένα σύμβολο κράτησης θέσης σχήματος [None, 512], θέλω να χωρίσει την tensor στην πρώτη dimenstion δυναμικά, με περίπου ακόμα τρόπο. Για παράδειγμα, εάν το σύμβολο κράτησης θέσης τροφοδοτείται με NumPy σχήμα [5, 512], θέλω να χωριστεί σε δύο τανυστές [2, 512]και [3, 512]. Αν το σύμβολο κράτησης θέσης τροφοδοτείται με NumPy σχήμα [7, 512], θέλω να χωριστεί σε δύο τανυστές [3, 512], [4, 512]. Υπάρχει κάποιος τρόπος να το κάνουμε αυτό; Η μπορεί να είναι χρήσιμο αν θέλετε να εφαρμόσετε τα δεδομένα παράλληλα σε πολλαπλές GPUs, ενώ το μέγεθος της παρτίδας είναι δυναμική.

Έχω προσπαθήσει να χρησιμοποιήσει tf.split( https://www.tensorflow.org/api_docs/python/tf/split ), αλλά δεν κατάφερε να κατασκευάσει η size_tensor.

Δημοσιεύθηκε 07/11/2018 στις 23:52
πηγή χρήστη
Σε άλλες γλώσσες...                            

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