ptxx_p的博客

pytorch的输入一般为batch，所以对于单独的输入(只有单个的输入！)来说，需要加上一维，具体操作为 a = torch.from_numpy(a).unsqueeze(dim = 0)需要注意的是，此时可能会报网络结构的错误，error提示如下：TypeError: conv2d() received an invalid combination of arguments - got (numpy.ndarray, Parameter, Parameter, tuple, tuple, tu.