如何将tensor大于某个值为1，小于某个值为0

最新推荐文章于 2023-03-02 22:29:11 发布
最新推荐文章于 2023-03-02 22:29:11 发布
阅读量1.1w 收藏 17
点赞数 8
分类专栏： python 文章标签： python tensor tensorflow where
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_32623363/article/details/103684766
版权

方法很简单，定义一个简单的函数，imgTensor是传入的需要转换的tensor，boundary是这个临界值，即大于boundary的值设为1，小于boundary的值设为0。本函数使用tf.where来实现，那么很容易可以看出我们可以任意设置tensor大于某个值（或小于某个值）时的值，只需要将下面代码中的tf.ones_like（或：tf.zeros_like）改成tf.fill()来构造一个相同大小的tensor即可。

def getBinaryTensor(imgTensor, boundary = 200):
    one = tf.ones_like(imgTensor)
    zero = tf.zeros_like(imgTensor)
    return tf.where(imgTensor > boundary, one, zero)
_吟游诗人
关注
  • 8
    点赞
  • 踩
  • 17
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
专栏目录
tensorflow 大于某个值为1,小于为0的实例
12-17
我就废话不多说了，大家还是直接看代码吧~ one = tf.ones_like(label) zero = tf.zeros_like(label) label = tf.where(label <0 x=zero, y=one) 补充知识：TensorFlow中获取大于零的元素集合 a为tensor idx> 0) output = tf.gather_nd(a, idx) 以上这篇tensorflow 大于某个值为1,小于为0的实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持软件开发网。
使用pytorch 筛选出一定范围的值
12-17
我就废话不多说了，大家还是直接看代码吧~ import torch input_tensor = torch.tensor([1,2,3,4,5]) print(input_tensor>3) mask = (input_tensor>3).nonzero() print(mask) print(input_tensor.index_select(0,mask)) tensor([0, 0, 0, 1, 1], dtype=torch.uint8) tensor([3, 4]) tensor([4, 5]) 补充知识：pytorch tensor筛选满足条件的行或列（使用与或） 我就废话不
将二维tensor矩阵中不为0元素转换为一维向量
zxs0222的博客
07-31 1773
将二维tensor矩阵中不为0元素转换为一维向量
tensorflow：大于阈值置1，小于置0
weixin_44613728的博客
04-17 6798
看了 UeFan写的tensorflow tf.where使用方法，大于某个值为1,小于为0文章，依然有些模糊，自己结合实际注解了一下。 a=tf.random.uniform([2,2]) #随机生成2*2矩阵a one = tf.ones_like(a) #生成与a大小一致的值全部为1的矩阵 zero = tf.zeros_like(a) l...
Pytorch 替换tensor中大于某个值的所有元素
进击的扛把子
01-21 6145
这是在自己做多标签分类任务时用到的，最后把结果sigmoid处理，值位于【0， 1】之间，大于阈值的取1，小于阈值的取0 两种方法： a = torch.rand((1, 3)) # 保证和a相同的维度大小 zero = torch.zeros_like(a) one = torch.ones_like(a) # a中大于0.5的用one(1)替换,否则a替换,即不变 a = torch.where(a > 0.5, zero, a) # a中小于0.5的用zero(0)替换,否则a替换
TensorFlow打印输出tensor的值
09-17
主要介绍了TensorFlow打印输出tensor的值，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
根据tensor的名字获取变量的值方式
09-18
本篇将介绍两种方法，通过tensor的名字来获取变量的值。 首先，让我们回顾一下示例中的代码。这里创建了一个随机输入`input`，然后通过`slim.fully_connected`函数构建了一个全连接层，这会隐式地创建权重`w`和偏置...
Pytorch 之修改Tensor部分值方式
12-23
比如输入全零tensor，可认为为黑色图片 >>> n=torch.FloatTensor(3,3,4).fill_(0) >>> n tensor([[[0., 0., 0., 0.], [0., 0., 0., 0.], [0., 0., 0., 0.]], [[0., 0., 0., 0.], [0., 0., 0., 0.], [0., 0.,...
tensorflow只保留张量top k的值不变，其余值变为0
李小喵的博客
08-15 2052
tensorflow中，只保留张量top k的值不变，其余值变为0，并且对剩余值进行归一化操作 比如：[0.2, 0.2, 0.4, 0.13, 0.07] ，获取top 3的值，但是保持原来向量形状，其余值设为0，结果为 [0.2, 0.2, 0.4, 0, 0]，若要用作权重，最好进行一下归一化操作，使得掩码后的向量和为1，[0.25, 0.25, 0.5, 0, 0]，这样便可以过滤掉一...
Pytorch.clamp：将小于0的元素修改为0，截断元素的取值空间
热门推荐
weixin_44633882的博客
04-18 1万+
Pytorch.clamp：将小于0的元素修改为0，截断元素的取值空间 torch.clamp(input, min, max, out=None) → Tensor 将输入中每个元素夹紧到[min,max]区间中，算法公式如下： yi={min⁡ if xi<min⁡xi if min⁡≤xi≤max⁡max⁡ if xi&gt...
tensor笔记2（tensor1==tensor2）
FUN6367的博客
07-24 480
代码： a = torch.tensor([1, 2, 3, 4, 5]) b = torch.tensor([1, 2, 2, 4, 6]) c = (a == b) print(c) 输出： tensor([ True, True, False, True, False])
把tensor 转换成0到1的值_C语言小写字母转换成大写字母
weixin_30990387的博客
01-20 957
2020.08.21C语言分享●●●#01#前言C语言案例---小写字母转换成大写字母从键盘上输入一个小写字母，按回车键，程序将该小写字母转换为大写字母，并输出其 ASCII 值。C语言案例---小写字母转换成大写字母1、利用 ASCII 码中大写字母和小写字母之间的转换关系(差值为 32)，可以将小写字母转换为大写字母。具体如下：2、利用 getchar 函数从标准输入里获取一...
关于Tensor的对象的“==”运算，发现“==”后可以使用sum()方法进行计数（即元素相等的个数）
qq_43570557的博客
01-21 1243
关于Tensor的对象的“==”运算，发现“==”后可以使用sum()方法进行计数（即元素相等的个数）
torch.max讲解-- 指定维度下，数值最大位置设为1其他为0
计算机视觉
11-20 3995
x是输入张量 dim指定维度 max表示取最大值 import torch if __name__ == '__main__': x = torch.randn([1, 3, 4, 4]).cuda() mask = (x == x.max(dim=1, keepdim=True)[0]).to(dtype=torch.int32) result = torch.mul(mask, x) print(x) print(mask) pri
控制Tensor里的数值小于0则置0
小白之比白更白的博客
03-02 1319
通用模板如下，第一个参数是控制条件，这里是小于0，第二个元素是满足条件的赋值，第三个是不满足条件的赋值，这里不小于0则保留原来的数值。同理，不小于0的可以赋值为1。
Pytorch基本数据操作（Tensor的运算）【深度学习从0到1】
AI_ayuan的博客
02-13 568
在PyTorch中，我们主要通过torch.Tensor对数据进行存储和变换操作，使用过numpy库的我们就可以发现，tensor和numpy里的多维数组十分相似。但是由于tensor提供GPU计算和自动求梯度等功能从而更适合深度学习。
【leetcode】73. 矩阵置零
Gin.TaMa的博客
07-29 199
给定一个 m x n 的矩阵，如果一个元素为 0，则将其所在行和列的所有元素都设为 0。请使用原地算法。 思路 想法 第二种方法不高效的地方在于我们会重复对同一行或者一列赋零。我们可以推迟对行和列赋零的操作。 我们可以用每行和每列的第一个元素作为标记，这个标记用来表示这一行或者这一列是否需要赋零。这意味着对于每个节点不需要访问 M+NM+N 个格子而是只需要对标记点的两个格子赋值。 if cel...
pytorch将小数tensor转换成整数tensor
咖乐部
09-14 5599
a = [[1.,2.],[3.,4.]] b = torch.tensor(a) # c = torch.tensor(b,dtype=torch.int) c = b.clone().type(torch.int) print(b) print(c)
pytorch 取出tensor中大于某个值的索引
最新发布
06-06
可以使用 pytorch 中的函数 `torch.where()` 来取出 tensor 中大于某个值的索引。具体步骤如下： 1. 首先使用 `torch.gt()` 函数来比较 tensor 中的每个元素是否大于目标值，返回一个 bool 类型的 tensor。 2. 然后...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值