/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

21 votes182 views3 answers
List of strings to numbers python
Python
4 votes188 views1 answer
Get os information python
Python
0 vote248 views1 answer
COLLECTING
Python
0 vote191 views1 answer
How to use drf pagination directly
Python
0 vote198 views1 answer
Print greeting in python explication
Python
0 vote181 views1 answer
F2 polar or nonpolar
Python
2 votes197 views3 answers
Python split string size
Python
0 vote223 views1 answer
How to convert linked list to list in python
Python
4 votes151 views4 answers
Fahrenheit to celsius python program
Python
0 vote225 views1 answer
Access field values of form django
Python Django