2016-06-06 51 views
0

我試圖乘以尺寸爲a = 40,000x1和b = 1x40,000的lua中的兩個矩陣。在Lua中,40,000x1矩陣顯示爲1D張量,1x40,000矩陣顯示爲2D張量。無論什麼時候,我嘗試使用a * b將它們相乘,出現錯誤:multiplication between 1D and 2D tensors not yet supported。我不能迭代地遍歷每個索引,因爲這個函數在我的程序中經常使用,並且會大大增加執行時間。我如何乘以ab一維和二維張量的Lua /火炬放大

回答

2

使用view

c = a:view(40000, 1) * b 
+0

錯誤:'期待一個連續的張量堆traceback'。 – Sibi

+0

'a:contiguous():view(40000,1)* b'​​(注意:這涉及到內存拷貝) – deltheil