随笔分类 - Pytorch
摘要:Pytorch中randn和rand函数的用法 randn torch.randn(\ sizes, out=None) → Tensor 返回一个包含了从 标准正态分布 中抽取的一组随机数的张量 size:张量的形状, out:结果张量。(目前还没有看到使用这个参数的例子) rand也差不多其实:
阅读全文
摘要:Pytorch中nn.Conv2d的用法 nn.Conv2d是二维卷积方法,相对应的还有一维卷积方法nn.Conv1d,常用于文本数据的处理,而nn.Conv2d一般用于二维图像。 先看一下接口定义: 参数解释: stride:步长 zero padding:图像四周填0 dilation:控制 k
阅读全文

浙公网安备 33010602011771号