Answers for "pytorch dataloader to numpy array"

Python

0

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

Posted by: Guest on October-24-2021

Code answers related to "pytorch dataloader to numpy array"

Code answers related to "Python"

Python Answers by Framework

Django
Flask

Browse Popular Code Answers by Language

Python

Javascript

Whatever

Shell/Bash

CSS

Html

PHP

SQL

Java

C#