tf.where 應(yīng)用到了張量的廣播機(jī)制　張量的廣播機(jī)制

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)

總結(jié)

以上是生活随笔為你收集整理的tf.where 用法的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。