/
githubmirror
/
keras
Обзор
Документация
Войти
/
githubmirror
/
keras
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
integration_tests/torch_workflow_test.py
34 строки
987 B
hertschuh
Fix order of arguments when calling `assertAlmostEqual` and `assertAllEqual`. (#22664)
15 апр 2026, 22:14
Не верифицирован
15 апр 2026, 22:14
a44f5be
Код
Авторство
О чём код?
import torch from keras.src import layers from keras.src import testing from keras.src.backend.common import KerasVariable class Net(torch.nn.Module): def __init__(self): super().__init__() self.fc1 = layers.Dense(1) def forward(self, x): x = self.fc1(x) return x class TorchWorkflowTest(testing.TestCase): def test_keras_layer_in_nn_module(self): net = Net() # Test using Keras layer in a nn.Module. # Test forward pass self.assertEqual(net(torch.empty(100, 10)).shape, (100, 1)) # Test KerasVariables are added as nn.Parameter. self.assertLen(list(net.parameters()), 2) # Test using KerasVariable as a torch tensor for torch ops. kernel = net.fc1.kernel transposed_kernel = torch.transpose(kernel, 0, 1) self.assertIsInstance(kernel, KerasVariable) self.assertIsInstance( torch.mul(kernel, transposed_kernel), torch.Tensor )