替换方法:
1.
matrix = tf.get_variable("Matrix", [output_size, input_size], dtype=input_.dtype)
=
matrix = Variable(torch.randn(output_size, input_size))
2.
self.D_l2_loss = tf.constant(0.0)
=
self.D_l2_loss = torch.Tensor(0.0)
3.
losses = tf.nn.softmax_cross_entropy_with_logits(logits=D_scores, labels=self.D_input_y)
=
losses = torch.nn.CrossEntropyLoss(D_scores, D_input_y)
4.
tf.reduce_mean(losses)
=
torch.mean(losses)
5.
embedded_chars = tf.nn.embedding_lookup(W_fe, Feature_input + 1)
=
embedded_chars = torch.index_select(W_fe, 0, Feature_input + 1)
6.
embedded_chars_expanded = tf.expand_dims(embedded_chars, -1)
=
embedded_chars_expanded = torch.unsqueeze(emb