/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

4 votes205 views1 answer
F-string ponto decimal python
Python
1 vote211 views1 answer
Django filter after annotate
Python Django
1 vote221 views1 answer
How to split channels wav python
Python
5 votes205 views4 answers
Python get last element in dict
Python
-1 vote192 views1 answer
Colours seaborn
Python
0 vote197 views1 answer
How to check if variable in python is of what kind
Python
0 vote207 views1 answer
Python regions
Python
1 vote198 views1 answer
Discord bot python on reaction
Python
51 votes173 views6 answers
Python print in color
Python
0 vote206 views1 answer
Ms access python dataframe
Python