/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

16 votes145 views3 answers
How to find if a value is even or odd in python
Python
0 vote119 views1 answer
Input function in django
Python
13 votes127 views2 answers
Only keep rows of a dataframe based on a column value
Python
1 vote195 views3 answers
Inheritance in python 3 example
Python
1 vote117 views1 answer
Pyhton built in sort
Python
0 vote132 views1 answer
AttributeError: 'Series' object has no attribute 'sort' site:stackoverflow.com
Python
-1 vote123 views2 answers
Jupyter matplotlib
Python
0 vote117 views2 answers
Local to ISO 8601 with TimeZone information (Python 3):
Python
0 vote212 views1 answer
Tf.slice
Python
17 votes127 views2 answers
How to record execution time in python
Python