src.utils.data_to_cudaΒΆ
Functions
Call cuda() on all tensor elements in inputs :param inputs: input list/dictionary :return: identical to inputs while all its elements are on cuda |
Functions
Call cuda() on all tensor elements in inputs :param inputs: input list/dictionary :return: identical to inputs while all its elements are on cuda |