- 博客(9)
- 收藏
- 关注
原创 【无标题】
import torch.nn.functional as F#输入:5x5input=torch.tensor([[1,2,0,3,1], [0,1,2,3,1], [1,2,1,0,0], [5,2,3,1,1], [2,1,0,1,1]])#卷积核3x3kernel=torch.tensor([[1,2,1],
2022-03-17 01:12:21
70
原创 【无标题】
from torch import nnimport torchclass Jiaxin(nn.Module): def __init__(self):#可以自己写也可以在Code-- generate--重写中选择 super(Jiaxin,self).__init__()#调用父类初始化函数 def forward(self,input): output=input+1#自定义 return output jiaxin=Jiax
2022-03-17 00:48:23
51
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人