该接口的目的:将自定义的Dataset根据batch size大小、是否shuffle等封装成一个Batch Size大小的Tensor,用于后面的训练。 dataset(Dataset): 传入的 ...
確定! 回上一頁