import torch
import torch.nn.functional as F
import math
n1 = torch.nn.Linear(2,4)
input1 = torch.randn(3,2)
input2 = torch.randn(2,3)
list1 = []
out1 = n1(input1)
out2 = F.relu(n1(input1))
print('out1 ',out1)
print('out1 size: ',out1.size())
print('out2 ',out2)
print('out2 size: ',out2.size())
list1.append(out1)
list1.append(out2)
print('input1: ',input1)
out3 = torch.stack(list1,1)
print('stack out3: ',out3)
print
pytroch常用函数
最新推荐文章于 2024-04-18 15:16:53 发布
![](https://img-home.csdnimg.cn/images/20240711042549.png)