1. 程式人生 > >tensorflow 怎麼對一個tensor進行所有元素相加求和

tensorflow 怎麼對一個tensor進行所有元素相加求和

tensor 和 numpy不同。
使用np.sum對一個tensor求和是不能得到一個標量的。

所以需要使用tensorflow自帶的函式對tensor進行求和。
總的來說,這裡我們需要明白一點,tensor資料使用tensorflow的函式運算,否則會出錯。

這裡僅給出對一個tensor進行所有元素相加求和的運算:
tf.reduce_sum(input)
這一行命令可以對input這個tensor進行元素相加。但是得到的仍然是tensor,並且也需要保證input是一個tensor