numpy数组上的pytorch.clamp()抛出TypeError

vngu2lb8 于 2023-03-08 发布在其他

关注(0)|答案(2)|浏览(146)

我有一张图片，它的背景不是黑色的，而它应该是黑色的。我在网上搜索寻找一个合适的策略来做到这一点，我找到了torch.clamp。我的图片如下：

当我使用以下命令时：

map = torch.clamp(map, Min=0)

我收到以下错误：

如何解决此问题？

2条答案

你需要传递给函数一个Tensor数组，而不是numpy数组，它在最后给出了如下的结果：

map_ = torch.clamp(torch.from_numpy(map_), min=0).numpy()

如果只使用了min参数，numpy.maximum可以做同样的事情：

>>> np.maximum(np.arange(-10, 10), 0)
array([0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9])