/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 vote129 views1 answer
Reversing Ints
Python
4 votes141 views3 answers
Update python ubuntu
Python
4 votes146 views2 answers
__call__ python
Python
1 vote135 views1 answer
Python hasattribute
Python
1 vote126 views1 answer
File to lowercase python
Python
66 votes97 views2 answers
'pip' is not recognized as an internal or external command, operable program or batch file.
Python
0 vote130 views1 answer
Pandas csv sum column
Python
0 vote128 views1 answer
List programs in python
Python
38 votes162 views2 answers
Delay time python
Python
0 vote135 views1 answer
Convert tensorflow checkpoint to pytorch
Python