生活随笔
收集整理的這篇文章主要介紹了
实验问题集中记录
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
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)
#待續🤪
總結
以上是生活随笔為你收集整理的实验问题集中记录的全部內容，希望文章能夠幫你解決所遇到的問題。
如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。
