Pytorch
来自Shiyin's note
跳到导航
跳到搜索
网络初始化
- 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,不能是其他