Python – tensorflow.boolean_mask()方法
TensorFlow是谷歌设计的开源Python库,用于开发机器学习模型和深度学习神经网络。 boolean_mask()是用于对Tensor应用布尔掩码的方法。
语法: tensorflow.boolean_mask(tensor, mask, axis, name)
参数:
- tensor: 它是一个N维的输入张量。
- mask: 它是一个具有k维度的布尔张量,其中k<=N,k是静态的。
- axis: 它是一个0维的张量,代表应用掩码的轴。轴的默认值是0,k+轴<=N。
- name: 这是一个可选的参数,定义了操作的名称。
返回:它返回(N-K+1)维的张量,这些张量的值与掩码中的真值相对应。
例子1:在这个例子中,输入是1-D。
输出:
例2:在这个例子中,采取的是2-D输入。
输出: