nn.Linear() 使用提醒

原本以为它是和nn.Conv2d()一样,就看第二个维度的数值,今天才知道,它是只看最后一个维度的数值!!!

例子1

python">'''
Descripttion: 
Result: 
Author: Philo
Date: 2024-02-27 14:33:50
LastEditors: Philo
LastEditTime: 2024-02-27 14:36:22
'''
import torch
import torch.nn as nn

# 定义输入特征数量和输出特征数量
in_features = 10
out_features = 5
# 创建一个线性变换层
linear_layer = nn.Linear(in_features, out_features)
# 随机生成一个输入张量
input_tensor = torch.randn(1, 12,32,10)
# 对输入张量进行线性变换
output_tensor = linear_layer(input_tensor)
# 打印输出张量
print("输出张量的形状:", output_tensor.shape)

输出张量的形状: torch.Size([1, 12, 32, 5])
在这里插入图片描述

例子2

在这里插入图片描述

例子3 Conv2d

在这里插入图片描述


http://www.niftyadmin.cn/n/5395981.html

相关文章

设计模式-结构型模式-适配器模式

0 引言 结构型模式(Structural Pattern)关注如何将现有类或对象组织在一起形成更加强大的结构。 1 适配器模式 适配器模式(Adapter Pattern):将一个接口转换成客户希望的另一个接口,使接口不兼容的那些类…

基于Springboot的计算机知识竞赛网站(有报告)。Javaee项目,springboot项目。

演示视频: 基于Springboot的计算机知识竞赛网站(有报告)。Javaee项目,springboot项目。 项目介绍: 采用M(model)V(view)C(controller)三层体系结…

Flink动态分区裁剪

1 原理 1.1 静态分区裁剪与动态分区裁剪 静态分区裁剪的原理跟谓词下推是一致的,只是适用的是分区表,通过将where条件中的分区条件下推到数据源达到减少分区扫描的目的   动态分区裁剪应用于Join场景,这种场景下,分区条件在joi…

腾讯云服务器4核8G性能,和阿里云比怎么样?

腾讯云4核8G服务器支持多少人在线访问?支持25人同时访问。实际上程序效率不同支持人数在线人数不同,公网带宽也是影响4核8G服务器并发数的一大因素,假设公网带宽太小,流量直接卡在入口,4核8G配置的CPU内存也会造成计算…

云端技术驾驭DAY13——Pod污点、容忍策略、Pod优先级与抢占、容器安全

往期回顾: 云端技术驾驭DAY01——云计算底层技术奥秘、云服务器磁盘技术、虚拟化管理、公有云概述 云端技术驾驭DAY02——华为云管理、云主机管理、跳板机配置、制作私有镜像模板 云端技术驾驭DAY03——云主机网站部署、web集群部署、Elasticsearch安装 云端技术驾驭…

面试题JS篇

目录 Js 基本数据类型有哪些Ajax 如何使用如何判断一个数据是 NaN?Js 中 null 与 undefined 区别闭包是什么?有什么特性?对页面会有什么影响JS中模块化的方法Js 中常见的内存泄漏什么是事件冒泡?事件委托是什么?如何确…

WPF 【十月的寒流】学习笔记(3):DataGrid分页

文章目录 前言相关链接代码仓库项目配置(省略)项目初始配置xamlviewModel Filter过滤详细代码展示结果问题 Linq过滤CollectionDataxamlviewModel sql,这里用到数据库,就不展开了 总结 前言 我们这次详细了解一下列表通知的底层是…

Redis 管道详解

Redis 管道 关键词:Pipeline Pipeline 简介 Redis 是一种基于 C/S 模型以及请求/响应协议的 TCP 服务。通常情况下,一个 Redis 命令的请求、响应遵循以下步骤: 客户端向服务端发送一个查询请求,并监听 Socket 返回&#xff08…