creating a tensor from a list of numpy.ndarray is extremely slow Please consider converting the lis
生活随笔
收集整理的這篇文章主要介紹了
creating a tensor from a list of numpy.ndarray is extremely slow Please consider converting the lis
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
我的代碼如下
node_2_neg_list = [torch.LongTensor(node) for node in node_2_negative]其中node_2_negative是一個list,里面有16個元素:
每個元素又是一個list,里面與10個元素:
而每個元素中又包含10個元素:
所以這是一個list嵌套list的情況,而我們執行上面的代碼,提示信息:
因此根據其實信息,我們需要將list先轉成numpy.ndarray然后再轉成tensor,原先的node_2_negative[0]數據如下:
然后轉成ndarray類型:
這樣就可以放心的把數據轉為tensor了:
總結
以上是生活随笔為你收集整理的creating a tensor from a list of numpy.ndarray is extremely slow Please consider converting the lis的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: DGL教程【五】使用自己的数据集
- 下一篇: 如何卸载流氓软件电脑如何卸载应用程序