2021年3月17日
PyTorch——Flatten（一）
摘要： 1 torch.nn.Flatten(start_dim=1, end_dim=-1) Parameters 1 start_dim – first dim to flatten (default = 1). 2 end_dim – last dim to flatten (default = -1 阅读全文
posted @ 2021-03-17 15:35 一剑光寒十四州 阅读(455) 评论(0) 推荐(0) 编辑
Pytorch——Linear(一）
摘要： 1 torch.nn.Linear(in_features, out_features, bias=True) 用于设置网络中的全连接层。 输入与输出均是二维张量。 输入与输出形状是 [batch_size, size]。 之前一般使用view或nn.Flatten()将4维张量变为2维张量。 Pa 阅读全文
posted @ 2021-03-17 14:52 一剑光寒十四州 阅读(602) 评论(0) 推荐(0) 编辑
PyTorch——池化（一）
摘要： torch.nn.MaxPool2d(kernel_size, stride=None, padding=0, dilation=1, return_indices=False, ceil_mode=False) Parameters 1 kernel_size – the size of the 阅读全文
posted @ 2021-03-17 14:29 一剑光寒十四州 阅读(181) 评论(0) 推荐(0) 编辑