/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 vote246 views1 answer
Python - Comment Parse String to List
Python
15 votes253 views3 answers
Send image discord.py
Python
1 vote310 views1 answer
On member leave event in discord.py
Python
0 vote262 views1 answer
Pyspark lit column
Python
0 vote269 views1 answer
How to read file again in python
Python
8 votes256 views3 answers
Sklearn roc curve
Python
0 vote278 views1 answer
Plotly dash datatable column width
Python
86 votes259 views5 answers
Print list in reverse order python
Python
0 vote262 views1 answer
How to get a summary of a column in python
Python
0 vote263 views2 answers
Python open google
Python