Что заставляет модуль nn Pytorch возвращать 1 или 0 для argmax этого массива?

#python #arrays #pytorch Вопрос: У меня есть массив, содержащий четыре случайных числа, которые функция argmax должна возвращать 0, 1, 2 или 3, но когда argmax вызывается из модели nn.Module, это…