import torch
from torch import nn
import numpy as np
class ScaledDotProductAttention(nn.Module)...手撕单头注意力机制
912 views
分类目录归档：算法手撕
手撕 numpy写线性回归的随机梯度下降
1024 views
import numpy as np
from matplotlib import pyplot as plt
# 生成权重以及偏执项layers_dim代表每层的神经元个数，
# 比如[...手撕 numpy写线性回归的随机梯度下降（stochastic gradient descent，SGD）
1084 views
# 数据加载
from sklearn.datasets import fetch_california_housing
from sklearn.model_selection import...