小编Ben*_*Ben的帖子

我想在二维数组上的 PyTorch 张量上做类似于 np.clip 的事情。更具体地说，我想将每一列剪辑在特定的值范围内（取决于列）。例如，在 numpy 中，您可以执行以下操作：

x = np.array([-1,10,3])
low = np.array([0,0,1])
high = np.array([2,5,4])
clipped_x = np.clip(x, low, high)

clipped_x == np.array([0,5,3]) # True

我找到了 torch.clamp，但不幸的是它不支持多维边界（整个张量只有一个标量值）。有没有一种“简洁”的方法可以将该功能扩展到我的案例中？

谢谢！