将张量中的数值调整到给定的范围内

我有一个 3D 张量，希望确保所有的值都在给定的范围内（在此情况下为 0-1）。为了做到这一点，我已经编写了以下代码：

    function capTo1or0 (Tensor3d)

       tensor_width=Tensor3d:size()[2]
       tensor_height=Tensor3d:size()[3]
       tensor_depth=Tensor3d:size()[1]
       for i=1,tensor_width,1 do
           for j=1,tensor_height,1 do
               for k=1,tensor_depth,1 do
                   if(Tensor3d[k][i][j])>1 then

                       Tensor3d[k][i][j]=1

                   end
                   if(Tensor3d[k][i][j]<0.0) then
                        Tensor3d[k][i][j]=0.0
                   end
               end
           end
        end
       return Tensor3d
    end

它可以工作，但是有一个问题：性能很差，我知道在手动循环数组之外大多数张量操作都要快得多，因此必须有某种更好的方法来做到这一点。 有人知道如何更快地做到这一点吗？

例如，我有一个 `2-3-3` 的数组，其值为

    [1,  2,  0.5][0.5,0.2,-0.2]
     [0.1,0.2,0.3][1,  1,   1  ]
    [-2, -1, 2  ][0.2,-5,-1   ]

那么我期望的结果是
    [1,  1,  0.5][0.5,0.2,0]
    [0.1,0.2,0.3][1,  1,   1  ]
    [0, 0, 1  ]  [0.2,0,-1   ]

将低于下限 0 的每个值替换为 0，将高于上限 1 的每个值替换为 1。

有人知道如何快速做到这一点吗？