0
点赞
收藏
分享

微信扫一扫

音视频入门基础:FLV专题(13)——FFmpeg源码中,解析任意Type值的SCRIPTDATAVALUE类型的实现

南陵王梁枫 2024-10-13 阅读 13

再接上文,补一下作者未补完的矩阵运算的坑。

首先贴一下原作者的图,将其转化为如下代码:

import torch
import torch.nn as nn
import torch.optim as optim

# 定义一个简单的两层神经网络
class TwoLayerNet(nn.Module):
    def __init__(self):
        super(TwoLayerNet, self).__init__()
        self.fc1 = nn.Linear(2, 2)  # 第一层
        self.fc2 = nn.Linear(2, 2)  # 第二层

    def forward(self, x):
        x = self.fc1(x)  # 第一层的前向传播
        x = self.fc2(x
举报

相关推荐

0 条评论