tf.keras.layers.UpSampling3D