2020-10-21 torch.Tensor.view()的用法

根據簡單有效原則,先上一張圖片:        圖中先給出一個名字爲C的Tensor,人爲賦值爲[ [ [ 1, 2, 3], [ 3, 4, 5 ] ], [ [ 6, 7, 8 ], [7, 8 , 9] ], [ [ 12, 34, 51 ], [ 78, 79, 80 ] ] ],在這一組簡單的數據中,batchsize=3即3個數據單元 [ [ 1, 2, 3], [ 3, 4, 5 ]
相關文章
相關標籤/搜索