Pycharm 中使用 net(X) 遇到的疑问

在学习 Pytorch 的过程中一直是使用 Jupyter notebook ,今天换到 Pycharm 中就发现了在 Pycharm 中使用 net(X) 不会输出前向计算的结果,只有使用 print(net(X)) 才会输出





import torch
from torch import nn

class MLP(nn.Module):

    def __init__(self, **kwargs):
       
        super(MLP, self).__init__(**kwargs)
        self.hidden = nn.Linear(784, 256) 
        self.act = nn.ReLU()
        self.output = nn.Linear(256, 10) 

    def forward(self, x):
        a = self.act(self.hidden(x))
        return self.output(a)
X = torch.rand(2, 784)
net = MLP()
print(net)
# print(net(X))
net(X)

img

如何才能使 net(X) 得到输出,因为在教程中

img


显示,该函数本来就能得到一个输出,如果不能输出,为什么

你这儿net(x)只是实例化MLP这个类,而且用的return返回值 没有print ,肯定没有是什么输出结果

out=net(X)
print(out)