Pytorch与Tensorflow的操作子对比
目录Linear
Linear
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', k
原创
2021-11-17 10:00:01 ·
1233 阅读 ·
0 评论