pytorch与tensorflow API速查表
pytorch与tensorflow API速查表
方法名称
pytroch
tensorflow
numpy
裁剪
torch.clamp(x, min, max)
tf.clip_by_value(x, min, max)
np.clip(x, min, max)
取最大值
torch.max(x, dim)[0]
tf.max(x, axis)
np.max(x, axi...
原创
2019-04-20 11:47:19 ·
7313 阅读 ·
0 评论