原本以為它是和nn.Conv2d()一樣,就看第二個維度的數值,今天才知道,它是只看最后一個維度的數值!!!
例子1
'''
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])