TensorFlow 2.0 RC1
import tensorflow as tf
from tensorflow.keras.models import Model
from tensorflow.keras.layers import Input, Multiply
import numpy as np
Expected output:
Multiply()([np.array([1,2,3,4,4,4]).reshape(2,3), np.array([1,0])])
Problem:
input_1 = Input(shape=(None,3))
mask_1 = Input(shape=(None,))
net = Multiply()([input_1, mask_1])
net = Model(inputs=[input_1, mask_1], outputs=net)
net.predict([np.array([1,2,3,4,4,4]).reshape(1,2,3), np.array([1,0]).reshape(1,2)]) # 1 = batch size
How to fix this problem?