torch.prod(input)返回input全部元素的乘積

import torch b = torch.tensor([2, 3.3]) print(torch.prod(b))

輸出
tensor(6.6000)

總結

以上是生活随笔為你收集整理的torch.prob的全部內容，希望文章能夠幫你解決所遇到的問題。

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