Python – método tensorflow.boolean_mask()

TensorFlow es una biblioteca Python de código abierto diseñada por Google para desarrollar modelos de aprendizaje automático y redes neuronales de aprendizaje profundo. boolean_mask() es un método utilizado para aplicar una máscara booleana a un tensor.

Sintaxis: tensorflow.boolean_mask(tensor, máscara, eje, nombre)

Parámetros:

• tensor: Es un tensor de entrada N-dimensional.
• mask: Es un tensor booleano con k-dimensiones donde k<=N y k se conoce estáticamente.
• eje: Es un tensor de dimensión 0 que representa el eje a partir del cual se debe aplicar la máscara. El valor predeterminado para el eje es cero y k+eje<=N.
• nombre: Es un parámetro opcional que define el nombre de la operación.

Retorno: Devuelve el tensor dimensional (N-K+1) que tiene los valores que se completan contra los valores verdaderos en la máscara. 
 

Ejemplo 1: En este ejemplo, la entrada es 1-D.

# importing the library
import tensorflow as tf
 
# initializing the inputs
tensor = [1,2,3]
mask = [False, True, True]
 
# printing the input
print('Tensor: ',tensor)
print('Mask: ',mask)
 
# applying the mask
result = tf.boolean_mask(tensor, mask)
 
# printing the result
print('Result: ',result)

Producción:

Tensor:  [1, 2, 3]
Mask:  [False, True, True]
Result:  tf.Tensor([2 3], shape=(2,), dtype=int32)

Ejemplo 2: En este ejemplo se toma una entrada 2-D.

# importing the library
import tensorflow as tf
 
# initializing the inputs
tensor = [[1, 2], [10, 14], [9, 7]]
mask = [False, True, True]
 
# printing the input
print('Tensor: ',tensor)
print('Mask: ',mask)
 
# applying the mask
result = tf.boolean_mask(tensor, mask)
 
# printing the result
print('Result: ',result)

Producción:

Tensor:  [[1, 2], [10, 14], [9, 7]]
Mask:  [False, True, True]
Result:  tf.Tensor(
[[10 14]
 [ 9  7]], shape=(2, 2), dtype=int32)