doing dynamic routing with tf.while_loop :arg proir: log prior for scaling with shape [10, num_caps] cap_prediction: predictions from layer below with shape [None, 10, num_caps, 16] num_caps: num_caps :return digit_caps: digit
(self, prior, cap_predictions, num_caps)
source not stored for this graph (policy: none)