prior_boxes = torch.FloatTensor(prior_boxes).to(device) # (8732, 4)
prior_boxes.clamp_(0, 1) # (8732, 4)
什么 dooes clamp_ 在 pytorch 中做什么以及如何将其更改为 tensorflow 2.0?
我不确定clamp_到底是做什么的?
prior_boxes = torch.FloatTensor(prior_boxes).to(device) # (8732, 4)
prior_boxes.clamp_(0, 1) # (8732, 4)
什么 dooes clamp_ 在 pytorch 中做什么以及如何将其更改为 tensorflow 2.0?
我不确定clamp_到底是做什么的?
clamp_(0, 1)
将所有元素夹在prior_boxes
[0, 1] 范围内。
张量流:
tf.clip_by_value
https://www.tensorflow.org/api_docs/python/tf/clip_by_value