class MolecularGCN(nn.Module): def __init__(self, in_feats, dim_embedding=128, padding=True, hidden_feats=None, activation=None): super(MolecularGCN, self).__init__() self.init_transform = nn.Linear(in_feats, dim_embedding, bias=False)是什么
时间: 2023-05-21 10:02:31 浏览: 181
get_feats:vgg16获得壮举
这是一个名为 MolecularGCN 的 PyTorch 模块,它是一个分子图卷积神经网络,用于分子结构的学习和预测。它的输入特征维度为 in_feats,输出特征维度为 dim_embedding,可以选择是否进行 padding,可以设置隐藏层特征维度和激活函数。其中,init_transform 是一个线性变换层,用于将输入特征转换为指定维度的特征表示。
阅读全文