1.torch.repeat和tf.tile
兩個函數都是將tensor的某一維度進行制定次數的重復：

a_ = torch.repeat(a, [x,y,x]) #a是三維tensor，將其1，2，3維分別repeat x, y, z次 b_ = tf.tile(b, [c,d,e,f]) #b是三維tensor，將其1，2，3，4維分別repeat a，b，c，d次

2.torch.diagonal和tf.diag_part
提取張量的主對角線元素

#torch x = torch.randn(4,4) tensor([[ 8.4527e-01, 1.9236e+00, -6.9886e-01, -4.1077e-01],[-1.4905e+00, 2.8126e-01, 9.9094e-02, -2.0024e-03],[-6.0617e-01, 1.0577e+00, -7.8548e-01, 2.0802e+00],[-7.9660e-01, 1.2871e+00, -3.5069e-01, -1.5513e+00]])torch.diagonal(x) # tensor([ 0.8453, 0.2813, -0.7855, -1.5513])#tensorflow a = tf.diag_part(b)

#待續🤪

總結

以上是生活随笔為你收集整理的实验问题集中记录的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。