摘要:
torch.cat 用于连接多个张量 当dim=0时张量按竖直方向连接,dim=1时按水平方向连接 阅读全文
摘要:
slice函数返回一列数。 返回的是数组,不包括end值。 阅读全文
摘要:
将数据集变成可以迭代的对象。 阅读全文
摘要:
可以让节点不进行求梯度,从而节省了内存控件,当神经网络较大且内存不够用时,就需要让梯度为False 代码: x = torch.tensor([1.0], requires_grad=True) with torch.no_grad(): y = x * 2 print(y.requires_gra 阅读全文
摘要:
使用不好就用相对路径,将程序和要读取的内容放在一起,之后读取时文件路径直接写文件名就可以 阅读全文