- 博客(10)
- 收藏
- 关注
RSS订阅
1原创 Xavier环境配置
此时已经安装完毕cuda和cudnn，并使用固态硬盘对xavier扩容，现在需要搭建pytorch环境，用于算法推理测试。
2024-07-02 16:18:18
1037
原创 【无标题】
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
127
原创 【无标题】
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
115
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人