pytorch学习笔记
athrunsunny
这个作者很懒,什么都没留下…
展开
-
将torch tensor转化为图像
反向操作transforms.Normalize和transforms.ToTensor函数,将tensor转换成.jpg、.png格式的图片# -*- coding: utf-8 -*-"""Time: 2022.01.17Author: AthrunsunnyVersion: V 0.1File: tensor_convert_to_img.pyDescribe: from torch tensor to image"""import tor...原创 2022-01-21 16:50:24 · 3466 阅读 · 0 评论 -
torch.gather的作用
总之,torch.gather()是一个非常有用的函数,可以实现许多gather、切片、重排类的操作。需要配合dim和index参数设定来达到不同的效果。这里沿着输入张量的第1轴(列轴)进行gather,索引index指定了每一行抽取的元素索引,最终得到新的输出张量。torch.gather()作为PyTorch函数,它可以从一个张量中按照指定的索引来抽取元素。原创 2023-10-07 10:18:34 · 199 阅读 · 0 评论 -
使用torch.nn.functional.pad填充图片
主要是想要给图像外围的上下左右填充像素点,通过实际的图像进行演示,通过opencv读取numpy.array形式的图像数据,而其中的维度分别表示图像的左右上下四个方向。具体的参数信息可以查看官方文档,这里就过多赘述。import torch.nn.functional as Fimport torchim = cv2.imread("demo.jpg")X=torch.tensor(im).transpose(0,2).transpose(2,1)...原创 2022-05-28 11:36:12 · 323 阅读 · 0 评论 -
理解torch.nn.Embedding
torch.nn.Embedding(num_embeddings: int, embedding_dim: int, padding_idx: Optional[int] = None, max_norm: Optional[float] = None, norm_type: float = 2., scale_grad_by_freq: bool = False, sparse: bool = False, _weight: Opti.原创 2022-02-22 15:15:39 · 1471 阅读 · 0 评论