摘要: ``` import torch import numpy as np import torch.nn as nn import torch.nn.functional as F import cv2 import matplotlib.pyplot as plt from PIL import Image class GaussianBlurConv(nn.Module): def __init 阅读全文
posted @ 2020-01-17 22:23 虔诚的树 阅读(3049) 评论(0) 推荐(0) 编辑
摘要: torch.squeeze 分为两种情况: 不指定维度 或 指定维度 1. 不指定维度 Example 2. 指定维度 & `output: (A, 1, B)` & `output: (A, B)` Example Note: The returned tensor shares the stor 阅读全文
posted @ 2020-01-17 13:50 虔诚的树 阅读(315) 评论(0) 推荐(0) 编辑