随笔分类 - 深度学习 / d2l
摘要：背景： 最近在学习d2l的时候，运行for X, y in train_iter:的时候，总是报错。查看报错的内容： RuntimeError: DataLoader worker (pid 41847) exited unexpectedly with exit code 1. Details a
阅读全文
摘要：一、降维 1、求和 对一个n维数组进行求和，如果指定沿哪一个轴来通过求和降低维度。如下是降低0维： A = torch.arange(20, dtype=torch.float32).reshape(5,4) print(A) print(A.sum(axis=0)) /* tensor([[ 0.
阅读全文
摘要：一、广播机制 两个张量进行运算（加减乘除幂等） 如果两个张量形状相同则很容易进行运算，如果两个张量不同的时候该如何进行运算呢？ 1、a张量竖为1，b张量横为1 比如：a = torch.arange(3).reshape((3, 1))， b = torch.arange(2).reshape((1
阅读全文
