この処理(言語で表せなかった）をtensorFlowからpytorchに移行したい

Question

この処理をtensorFlowから
pytorchでも動かせるようにしたいです

tenosrFlow版

        l0 = Dense(int(self.actions_dim * (self.actions_dim + 1) / 2), activation="linear", name='l0')(h)
        l1 = Lambda(lambda x: tf.contrib.distributions.fill_triangular(x))(l0)
        L = Lambda(lambda x: tf.matrix_set_diag(x, tf.exp(tf.matrix_diag_part(x))))(l1)

これが
[1,1,1,1,1,1]

こうなれば成功のはず（tensorflowバージョン変えたせいでためせない）
[e,0,0]
[1,e,0]
[1,1,e]

e=expです

tf.matrix_set_diag(x
tf.contrib.distributions.fill_triangular(x)
pytorchでこれにあたる関数はありませんか？

あとこの処理　を言語かできる人いたら教えてください