【问题标题】:is there any similar function with clamp_ in tensorflow > 2.0在 tensorflow > 2.0 中是否有与 clamp_ 类似的功能
【发布时间】:2020-06-02 00:40:26
【问题描述】:

我正在将 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)?

【问题讨论】:

    标签: tensorflow pytorch tensorflow2.0 clamp


    【解决方案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.]]
    

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2016-05-31
      • 2012-08-01
      • 2023-04-07
      相关资源
      最近更新 更多