Python tensorflow.boolean_mask()用法及代码示例

TensorFlow是Google设计的开源Python库，用于开发机器学习模型和深度学习神经网络。 boolean_mask()是用于将布尔蒙版应用于张量的方法。

用法：tensorflow.boolean_mask(tensor, mask, axis, name)

参数：

tensor:这是一个N-dimensional输入张量。
mask:这是一个具有k-dimensions的布尔张量，其中k <= N并且k是静态已知的。
axis:这是一个0维张量，用于重新设置应从其应用蒙版的轴。轴的默认值为零，并且k + axis <= N。
name:这是一个可选参数，用于定义操作的名称。

返回：它返回(N-K + 1)维张量，其中的值相对于mask中的True值填充。

范例1：在此示例中，输入为一维。

Python3

# 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)

输出：

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

范例2：在此示例中，采用了二维输入。

Python3

# 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)

输出：

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

相关用法

注：本文由纯净天空筛选整理自aman neekhara大神的英文原创作品 Python – tensorflow.boolean_mask() method。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。