TensorFlow函數(shù):tf.losses.cosine_distance

2018-08-20 11:24 更新

tf.losses.cosine_distance函數(shù)

tf.losses.cosine_distance(
    labels,
    predictions,
    axis=None,
    weights=1.0,
    scope=None,
    loss_collection=tf.GraphKeys.LOSSES,
    reduction=Reduction.SUM_BY_NONZERO_WEIGHTS,
    dim=None
)

定義在:tensorflow/python/ops/losses/losses_impl.py.

在訓(xùn)練過(guò)程中增加了余弦距離loss.(已否決的參數(shù))

某些參數(shù)已棄用.它們將在未來(lái)版本中刪除.更新說(shuō)明:不推薦使用dim,而是使用axis

請(qǐng)注意,該函數(shù)假定predictions和labels已經(jīng)進(jìn)行了單元規(guī)范化.

參數(shù):

  • labels:形狀與“predictions”匹配的Tensor
  • predictions:任意矩陣.
  • axis:計(jì)算余弦距離的維數(shù).
  • weights:可選,秩為0或與labels具有相同秩的Tensor,并且必須可廣播到labels(即,所有維度必須為1,或者與相應(yīng)的losses的維度相同).
  • scope:計(jì)算loss時(shí)執(zhí)行的操作范圍.
  • loss_collection:將添加此loss的集合.
  • reduction:適用于loss的減少類型.
  • dim:axis的舊的(已棄用)名稱.

返回:

加權(quán)損耗浮動(dòng)張量.如果還原為 NONE, 則與標(biāo)簽的形狀相同;否則, 它是標(biāo)量.


加權(quán)損失浮動(dòng)Tensor.如果reduction是NONE,則它的形狀與labels相同;否則,它是標(biāo)量.

可能引發(fā)的異常:

  • ValueError:如果predictions的形狀與labels的形狀不匹配,或者axis,labels,predictions、weights是None.
以上內(nèi)容是否對(duì)您有幫助:
在線筆記
App下載
App下載

掃描二維碼

下載編程獅App

公眾號(hào)
微信公眾號(hào)

編程獅公眾號(hào)