PyTorch 按索引赋值

https://blog.csdn.net/u012759262/article/details/103122618

PyTorch

   import torchx_dat = torch.tensor([[1, 2, 9], [4, 7, 6], [7, 5, 3]], dtype=torch.float)index = torch.tensor([0, 2])x_dat.index_fill_(0, index, -1)#按行填充，填0和2行# x_dat.index_fill_(1, index, -1)#按列填充，填0和2列print(x_dat)# 在列的维度(dim=0)，取每列的最大值overlap_for_each_prior, object_for_each_prior = x_dat.max(dim=0)#按列max_object, prior_for_each_object = x_dat.max(dim=1)  # 每行的最大值print('max_prior',overlap_for_each_prior)print('max_object',max_object)print('++++++++++++++++++')print(object_for_each_prior)print(prior_for_each_object)#####################################################################################################'''这里是关键'''print(object_for_each_prior[prior_for_each_object])

PyTorch 按索引赋值相关推荐

pytorch之torch.zeros_like，torch.ones_like和tensor按索引赋值
1. torch.zeros_like 可以生成和括号内变量维度一致的全是0的内容.例如:输入维度为 1 × 8 × 3 × 5 1\times8\times3\times5 1×8×3×5的x,要生 ...
pytorch之tensor按索引赋值，三种方法！
这里,我们介绍其中一种方法,即torch.scatter_()函数 import torch label = torch.zeros(3, 6) #首先生成一个全零的多维数组 print(" ...
pytorch中谜一样的索引赋值法
test import torch nums = 3 # 首先,我们随机生成一个3*5的矩阵 a = torch.randn(nums, 5) # 在列的维度(dim=0),取每列的最大值 overl ...
pytorch问题索引
20210623 RuntimeError: expected device cpu and dtype Float but got device cpu and dtype Bool () 调试到对 ...
pytorch的索引与切片
** 一.从左边开始索引 ** In [12]: a = torch.rand(4,3,28,28)In [13]: a.shape Out[13]: torch.Size([4, 3, 28, 28 ...
pytorch按照索引取batch中的数
比如说bert的输出表征是基于子词的,想要用于基于词的任务,需要将词对应的最后一个子词的表征取出来,代码如下: import torch import torch.nn as nntorch.manu ...
第二课.PyTorch入门
什么是PyTorch Pytorch是一个基于Python的科学计算库,类似于Numpy,但可以使用GPU,可灵活开展深度学习实验,其数据结构tensor类似于ndarray,但tensor可以在GP ...
Pytorch学习笔记总结
往期Pytorch学习笔记总结: 1.Pytorch实战笔记_GoAI的博客-CSDN博客 2.Pytorch入门教程_GoAI的博客-CSDN博客 Pytorch系列目录: PyTorch学习笔记( ...
从基础概念到实现，小白如何快速入门PyTorch
转自:https://blog.csdn.net/Julialove102123/article/details/80487269 PyTorch中文文档官网教材:https://pytorch.o ...

PyTorch 按索引赋值

PyTorch 按索引赋值相关推荐

最新文章

热门文章