не реализована реализация сети капсул tenorflow 2 - PullRequest
Новая
       6

не реализована реализация сети капсул tenorflow 2

0 голосов
/ 04 февраля 2020

Я работаю над реализацией капсульной сети в TensorFlow version-2-gpu. в то время как я делаю изменение формы на выходе слоя свертки (тензор), это дает ошибку при попытке преобразовать значение. ошибка и мой код, как показано ниже. 

 conv1_params = {"filters": 256,"kernel_size": 9,"strides": 1,"padding": 
                  "valid","activation":tf.nn.relu,}
 conv2_params = {"filters": caps1_n_maps * caps1_n_dims,"kernel_size":9,"strides": 2,"padding": 
                 "valid","activation": tf.nn.relu}
 conv1 = tf.keras.layers.Conv2D(input_shape=(None,28,28,1), name="conv1", **conv1_params)
 conv2 = tf.keras.layers.Conv2D(name="conv2", **conv2_params)
 #output shape of conv1=TensorShape([None, 20, 20, 256])
 #output shape of conv2=TensorShape([None, 6, 6, 256])
 caps1_raw=tf.keras.backend.reshape(conv2,shape=[-1,caps1_n_caps,caps1_n_dims])

ошибка

ValueError: Попытка преобразовать значение () с неподдерживаемым типом () в Tensor.

...