pytorch
ztono
某大厂后端工程师
展开
-
pytorch入门之Tensor和Variable
pytorch入门之Tensor tensor是什么 Tensor译为张量。我的理解Tensor和多维数组比较相似。当维度为0时就是常量,为1时是向量。和Tensorflow一样,pytorch也使用torch作为数据格式。 在pytorch中,tensor可以和ndarray(numpy中)相互转换,而且pytorch中的tensor可以方便的放到GPU上进行计算。 使用tensor pytor...原创 2019-05-15 01:14:19 · 2537 阅读 · 0 评论 -
pytorch之自定义数据集
pytorch之自定义数据集 pytorch提供了torch.utils.data.Dataset这一抽象类来定义自己的数据集。使用时需要定义__len__ 和__getitem__这两个函数。 例如,定义一个最简单的数据集 class myDataset(Dataset): def __init__(self, file_path): self.csv_data = p...原创 2019-05-15 23:45:24 · 937 阅读 · 0 评论