目录
Pytorch
torch.nn.Linear(in_features, out_features,
bias=True, device=None, dtype=None)
Tensorflow
tf.keras.layers.Dense(
units, activation=None, use_bias=True,
kernel_initializer='glorot_uniform',
bias_initializer='zeros', kernel_regularizer=None,
bias_regularizer=None, activity_regularizer=None, kernel_constraint=None,
bias_constraint=None, **kwargs
)
结果不同的原因是权重和偏移的张量不同。