torch实现clip by tensor操作

tf的clip_by_value不仅可以clip by number还能clip by tensor,到了torch暂时没有找到，只有clamp，其中max和min必须是number。

自己实现了一下。

def clip_by_tensor(t,t_min,t_max):"""clip_by_tensor:param t: tensor:param t_min: min:param t_max: max:return: cliped tensor"""t=t.float()t_min=t_min.float()t_max=t_max.float()result = (t >= t_min).float() * t + (t < t_min).float() * t_minresult = (result <= t_max).float() * result + (result > t_max).float() * t_maxreturn result

使用方法：

b=torch.randint(0,10,(3,3))
print(b)
min=torch.randint(0,5,(3,3))
print(min)
max=torch.randint(6,10,(3,3))
print(max)
print(clip_by_tensor(b,min,max))

输出结果：

tensor([[2, 8, 8],
[3, 1, 6],
[2, 4, 4]])
tensor([[3, 0, 3],
[2, 4, 0],
[0, 3, 2]])
tensor([[9, 8, 7],
[9, 8, 6],
[9, 7, 9]])
tensor([[3., 8., 7.],
[3., 4., 6.],
[2., 4., 4.]])

torch实现clip by tensor操作相关推荐

pytorch tensor操作：tensor与numpy转换
tensor转numpy t = torch.ones(5) print(f"t: {t}") n = t.numpy() print(f"n: {n}") 输 ...
PyTorch: tensor操作(一) contiguous
本文目录 tensor在内存中的存储信息区和存储区 shape && stride contiguous 什么时候用contiguous呢? 为什么要用contiguous 为什么c ...
pytorch Tensor操作（二）
高级索引 PyTorch在0.2版本中完善了索引操作,目前已经支持绝大多数numpy的高级索引.高级索引可以看成是普通索引操作的扩展,但是高级索引操作的结果一般不和原始的Tensor贡献内出. x = ...
python选取tensor某一维_Pytorch的Tensor操作(1)
类型推断 torch.randn():随机初始化 a.type():返回类型 type():返回基本类型 isinstance() :检查类型 cuda会影响数据类型标量维度(dimention) ...
torch将多个tensor张量合并为一个张量，只提高迷你批次的纬度
在自己写pytorch的数据集加载函数时,会遇到一个问题,如何将多张图片张量合并到一起,提升迷你批次batch的纬度,但是不影响图片的大小和RGB通道数解决方法: 函数torch.cat(input ...
pytorch之tensor操作expand
import torch# expand 将tensor按照某一个维度扩大或者将tensor整体扩大 # expand 不会额外分配空间,只是原tensor的一个视图x = torch.tensor( ...
torch对于tensor的常规操作
前言使用pytorch框架,会常操作tensor,以下则是对tensor常规操作的汇总. import torch torch.Tensor会继承某些torch的某些数学运算,例如sort, min ...
torch.Tensor常用数据操作汇总与自动求梯度
在深度学习中,通常会频繁地对数据进行操作,在PyTorch中,torch.Tensor是存储和变换数据的主要工具(Tensor和NumPy的多维数组非常类似,Tensor提供GPU计算和自动求梯度等更 ...
【PyTorch系例】torch.Tensor详解和常用操作
学习教材: 动手学深度学习 PYTORCH 版(DEMO) (https://github.com/ShusenTang/Dive-into-DL-PyTorch) PDF 制作by [Marcus ...

torch实现clip by tensor操作

torch实现clip by tensor操作相关推荐

最新文章

热门文章