我正在将 Torch 代码转换为 tensorflow 2.0
prior_boxes = torch.FloatTensor(prior_boxes).to(device) # (8732, 4)
prior_boxes.clamp_(0, 1) # (8732, 4)
在 tensorflow > 2.0 中是否可以替换clamp_(0,1)？
user11173832
问问题
121 次
1 回答
1
尝试tf.clip_by_value，虽然不像clamp_，但它不是就地的：
t = tf.constant([[-10., -1., 0.], [0., 2., 10.]])
t2 = tf.clip_by_value(t, clip_value_min=-1, clip_value_max=1)
t2.numpy()
# gives [[-1., -1., 0.], [0., 1., 1.]]
于 2020-06-02T00:48:32.267 回答