怎么在Linear里设置因变量取值只能为整数

可以使用 torch.nn.functional.relu() 函数将因变量的值取整。具体实现代码如下所示：

import torch.nn as nn
import torch.nn.functional as F

class LinearInt(nn.Module):
    def __init__(self, in_features, out_features, bias=True):
        super(LinearInt, self).__init__()
        self.in_features = in_features
        self.out_features = out_features
        self.weight = nn.Parameter(torch.Tensor(out_features, in_features))
        if bias:
            self.bias = nn.Parameter(torch.Tensor(out_features))
        else:
            self.register_parameter('bias', None)
        self.reset_parameters()

    def reset_parameters(self):
        nn.init.kaiming_uniform_(self.weight, a=math.sqrt(5))
        if self.bias is not None:
            fan_in, _ = nn.init._calculate_fan_in_and_fan_out(self.weight)
            bound = 1 / math.sqrt(fan_in)
            nn.init.uniform_(self.bias, -bound, bound)

    def forward(self, input):
        output = F.linear(input, self.weight, self.bias)
        return F.relu(output).round_()

上述代码中，LinearInt 是一个继承自 nn.Module 的自定义层，其 forward() 函数对输出结果应用了 relu() 函数，从而将因变量的值限定为整数。使用该层可以实现在 Linear 层中设置因变量取值只能为整数的功能。