我有一個張x和x.shape=(batch_size,10)，現在我想利用 x[i][0] = x[i][0]*x[i][1]*...*x[i][9] for i in range(batch_size)這是我的代碼：for i in range(batch_size): for k in range(1, 10): x[i][0] = x[i][0] * x[i][k]但是當我實現這個forward()并調用時loss.backward()，反向傳播的速度非常慢。為什么它很慢，有什么方法可以有效地實現它？
如何有效地計算 Pytorch 中的張量？
慕婉清6462132
2021-09-11 20:07:20
