记录之tensorflow和pytorch中的取范数归一化操作
生活随笔
收集整理的這篇文章主要介紹了
记录之tensorflow和pytorch中的取范数归一化操作
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
最近在將tensorflow翻譯成torch的時候遇到了一個問題,就是有一個函數 tf.nn.l2_normalize沒有找到相應的替換函數,無奈自己手寫了一個這樣的操作,最近在閱讀令一個torch源碼時發現了,就及時記錄
tf:
import tensorflow as tftf.nn.l2_normalize(x, dim, epsilon=1e-12, name=None)參數:
x:輸入;?
dim:為l2范化的維數
epsilon:范化的最小值邊界
torch:
import torch.nn.functional as FF.normalize(input, p=2, dim=1, eps=1e-12, out=None)參數:
input:輸入
p:范數類型
dim:維度
eps:范化的最小值邊界
總結
以上是生活随笔為你收集整理的记录之tensorflow和pytorch中的取范数归一化操作的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: AssertionError: Inva
- 下一篇: tensorflow实现梯度累计,再回传