/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

2 votes189 views1 answer
Extract text regex python
Python
0 vote190 views1 answer
Truncate string python
Python
0 vote186 views1 answer
Python selenium not returning correct source
Python
0 vote213 views2 answers
Method get first last name python
Python
0 vote181 views1 answer
Python textbox
Python
2 votes218 views2 answers
Range python start at 1
Python
14 votes228 views1 answer
2 list difference python
Python
0 vote194 views1 answer
Find type of an element in list python
Python
0 vote188 views1 answer
Python check column conditions
Python
2 votes196 views3 answers
Split string by length python
Python