def normalize(w):
"""Normalizes weight tensor over full filter."""
return F.normalize(w.view(w.size(0), -1)).view_as(w)
微信
公众号
扫码关注公众号
微信
公众号