/Python/Pytorch dataloader to numpy array

Pytorch dataloader to numpy array

# index is the specific data your want in dataloader
import numpy as np
def transfer_dataloader(dataloader, index=0):
    cache_list = list(iter(dataloader))
    assert len(cache_list) > 0
    assert index < len(cache_list[0])
    result_list = np.array(list(map(lambda x: x[index].numpy(), cache_list)))
    return result_list

Similar codes

0 vote145 views10 answers
Convert python code to c++ online
Python
0 vote180 views2 answers
Wireframes and Surface Plots
Python
-1 vote190 views1 answer
Uninstall python3.8 ubuntu
Python
3 votes149 views2 answers
How to plotting bar on matplotlib
Python
33 votes159 views6 answers
Python if else one line
Python
3 votes156 views2 answers
Import class from another file python
Python
11 votes155 views6 answers
How to add to a list python
Python
1 vote157 views2 answers
Why use python
Python
0 vote139 views1 answer
Python print error output
Python
0 vote158 views1 answer
Django filter word count greater than
Python