import numpy as np
a = np.random.randint(-5, 5, (1, 10))
c=np.sum(a>=1) #條件為大於等於1
print ("隨機數組a : "+str(a)) #輸出數組a
print ("大於等於1的個數: "+str(c)) #輸出滿足條件的個數
np.sum((log_prob>-0.2).cpu().numpy().astype(int)) > 0 #把tensor 轉為numpy,之后再統計array中大於-0.2的元素的個數