Update inference-tensorflow.R

Alteração sugerida em Irrationone#92 (comment) para corrigir erro de execução
GabrielPlaza95 · Mar 14, 2022 · 5f56825 · 5f56825
1 parent be66325
commit 5f56825
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/inference-tensorflow.R b/R/inference-tensorflow.R
@@ -162,7 +162,7 @@ inference_tensorflow <- function(Y,
 
   Q = -tf$einsum('nc,cn->', gamma_fixed, p_y_on_c_unorm)
 
-  p_y_on_c_norm <- tf$reshape(tf$reduce_logsumexp(p_y_on_c_unorm, 0L), shape(1,-1))
+  p_y_on_c_norm <- tf$reshape(tf$reduce_logsumexp(p_y_on_c_unorm, 0L), as_tensor(shape(1,NULL)))
 
   gamma <- tf$transpose(tf$exp(p_y_on_c_unorm - p_y_on_c_norm))