TensorFlow转pytorch

Question

用TensorFlow的函数怎么能转成pytorch的！！好难转，而且有很多错误。帮忙转一下以下三个，调试没问题就采纳。
batch_size = tf.placeholder(tf.int32, shape=()) placeholder这个函数怎么改！！
s = tf.random_uniform(shape=[batch_size], minval=0, maxval=self.M, dtype=tf.int64)
cross_entropy = tf.losses.sparse_softmax_cross_entropy(labels=s, logits=s_hat)