feat(examples): 更新CNN示例中的全连接层实现 - 修复权重矩阵大小计算,根据展平后的实际大小动态创建权重 - 添加张量重塑逻辑,将展平后的张量转换为正确的2D格式进行矩阵乘法 - 使用动态生成的权重值替代固定的权重数组 - 确保矩阵乘法维度匹配:(1, N) * (N, 2) = (1, 2) ``` |
||
|---|---|---|
| .. | ||
| autograd | ||
| basic_operation | ||
| linear_regression | ||
| cnn_example.go | ||
feat(examples): 更新CNN示例中的全连接层实现 - 修复权重矩阵大小计算,根据展平后的实际大小动态创建权重 - 添加张量重塑逻辑,将展平后的张量转换为正确的2D格式进行矩阵乘法 - 使用动态生成的权重值替代固定的权重数组 - 确保矩阵乘法维度匹配:(1, N) * (N, 2) = (1, 2) ``` |
||
|---|---|---|
| .. | ||
| autograd | ||
| basic_operation | ||
| linear_regression | ||
| cnn_example.go | ||