首页 > 其他分享 >18-神经网络-自定义带参数的层

18-神经网络-自定义带参数的层

时间:2024-08-25 11:49:29浏览次数:8  
标签:__ nn 自定义 18 self torch 神经网络 units linear

1、nn.Parameter函数


2、torch.mm 和torch.matmul区别
都是 PyTorch 中用于矩阵乘法的函数,但它们在使用上有细微的差别

import torch
import torch.nn as nn
import torch.nn.functional as F

class MyLinear(nn.Module):
    def __init__(self, in_units, out_units):
        super(MyLinear, self).__init__()
        self.weight = nn.Parameter(torch.randn((in_units, out_units)))
        self.bias = nn.Parameter(torch.randn(out_units))

    def forward(self, x):
        linear = torch.matmul(x, self.weight) + self.bias
        return F.relu(linear)

linear = MyLinear(5, 3)
print(linear.weight)

y = linear(torch.rand((2, 5)))
print(y)

标签:__,nn,自定义,18,self,torch,神经网络,units,linear
From: https://www.cnblogs.com/morehair/p/18378769

相关文章

  • apt update 报错:Could not handshake: Error in the pull function. [IP: 185.199.108
    sudoaptupdate报错:错误:12https://nvidia.github.io/nvidia-container-runtime/stable/ubuntu18.04/amd64ReleaseCouldnothandshake:Errorinthepullfunction.[IP:185.199.108.153443]错误:13https://nvidia.github.io/nvidia-docker/ubuntu18.04/amd64Re......
  • 17-神经网络-延迟初始化
    使用torch.nn.LazyLinear(output)实现延迟初始化importtorchimporttorch.nnasnnclassMyModel(nn.Module):def__init__(self):super(MyModel,self).__init__()self.fc1=nn.LazyLinear(128)#输入维度设置为None,表示延迟初始化self......
  • 14-神经网络-参数管理
    torch.rand和torch.randn的区别:......
  • 18 个一线工作中常用 Shell 脚本
    1、检测两台服务器指定目录下的文件一致性2、定时清空文件内容,定时记录文件大小3、检测网卡流量,并按规定格式记录在日志中4、计算文档每行出现的数字个数,并计算整个文档的数字总数5、从FTP服务器下载文件6、连续输入5个100以内的数字,统计和、最小和最大7、监测Nginx访......
  • SpringBoot自定义校验
    通常情况,后端在业务层需要进行大量校验,写在业务层又不美观,而且需要重复编写,很是不方便,Spring提供的校验注解有时无法满足我们的需求,需要自定义校验规则,以校验手机号为例,下面开始自定义校验一、引入依赖引入Spring校验依赖包<dependency>  <groupId>org.springframework......
  • 《探秘神经网络:人工智能的强大引擎》
    在当今科技飞速发展的时代,人工智能已经成为了热门话题,而神经网络作为人工智能的重要组成部分,正发挥着越来越关键的作用。那么,什么是神经网络呢?它在人工智能中又有哪些令人瞩目的应用呢?一、什么是神经网络神经网络,也被称为人工神经网络,是一种模仿生物神经网络结构和功能的......
  • GNN的理解难点:一种不同于传统神经网络的复杂性
    图神经网络(GNN)已经成为深度学习领域的一颗新星,它在处理图形数据方面显示出了巨大的潜力和优势。然而,许多研究者和开发者发现GNN比传统的神经网络更难以理解和掌握。本文将探讨GNN的理解难点,以及它与传统神经网络在概念和实现上的主要差异。一、图数据的复杂性首先,GNN之......
  • 图神经网络GNN的前世今生
    GNN图神经网络(GraphNeuralNetwork,简称GNN)已经成为处理图形结构数据的一种强大工具,广泛应用于社交网络分析、知识图谱、推荐系统等领域。在本文中,我们将深入探讨图神经网络的历史背景、关键的发展阶段以及未来可能的发展方向。一、背景介绍图(Graph)是一种数据结构,......
  • 【Material-UI】深入探讨Radio Group组件的自定义功能
    文章目录一、RadioGroup组件概述1.组件介绍2.自定义的重要性二、RadioGroup组件的自定义1.样式定制示例2.代码详解3.样式自定义的注意事项三、如何利用自定义功能提升用户体验1.提升视觉一致性2.增强可用性3.实现更灵活的布局四、总结Material-UI是R......
  • 使用Java导出MySQL数据:支持自定义分隔符的TXT文件生成
    在软件开发中,我们经常需要将数据库中的数据导出为文件,给关联系统做传输或者进行数据备份、迁移或分析。常见的导出格式包括CSV和TXT文件,分隔符可以是逗号、制表符或其他符号。本文将深入探讨如何使用Java从MySQL数据库中导出数据,并支持用户自定义分隔符来生成格式化的TXT文件。......