tf.where 應用到了張量的廣播機制　張量的廣播機制

tf.where(condition, x=None, y=None, name=None )

[if condition[k] =True 取　x[k] else 取　y[k]]

import tensorflow as tf c=tf.where([True, False, False, True], [1,2,3,4], [100,200,300,400]) print(c)c=tf.where([True, False, False, True], [1,2,3,4], [100]) print(c)c=tf.where([True, False, False, True], [1,2,3,4], 100) print(c)c=tf.where([True, False, False, True], 1, 100) print(c) tf.Tensor([ 1 200 300 4], shape=(4,), dtype=int32) tf.Tensor([ 1 100 100 4], shape=(4,), dtype=int32) tf.Tensor([ 1 100 100 4], shape=(4,), dtype=int32) tf.Tensor([ 1 100 100 1], shape=(4,), dtype=int32)

總結

以上是生活随笔為你收集整理的tf.where 用法的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。