Pytorch

来自Shiyin's note
Shen讨论 | 贡献2021年10月16日 (六) 10:32的版本
跳到导航 跳到搜索

网络初始化

  • Xavier and Kaiming initialization [4]

函数

  • torch.clamp(input, min=None, max=None, *, out=None) → Tensor
Clamps all elements in input into the range [ min, max ]. Letting min_value and max_value be min and max, respectively

Tensor

  • cpu() numpy() detach() item() [5]
注意cuda上面的变量类型只能是tensor,不能是其他