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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。