Задача курса
Умножение матриц NumPy через @: расчёт предсказаний
Курс «NumPy и pandas: практический тренажёр» · урок «Матрично-векторные произведения через оператор @»
Закрепите порядок измерений при умножении матрицы признаков на матрицу весов.
Условие
Матрица весов и матрица объектов: предсказания через @
Что уже дано
Проверяющая система запускает вашу программу и подаёт данные через stdin. Никаких заранее созданных переменных нет — вы читаете всё сами.
Что нужно сделать
Напишите программу, которая:
- Читает из
stdinтри целых числа в одной строке:M,K,N— гдеM— число объектов,K— число признаков,N— число выходов. - Читает следующие
Mстрок; каждая содержит ровноKвещественных чисел через пробел. Создаёт двумерный массивXчерезnp.array(..., dtype=float)формы(M, K)— матрица объектов (строка = один объект). - Читает следующие
Kстрок; каждая содержит ровноNвещественных чисел через пробел. Создаёт двумерный массивWчерезnp.array(..., dtype=float)формы(K, N)— матрица весов. - Вычисляет матрицу предсказаний
Y = X @ Wоператором@. ФормаYдолжна быть(M, N). - Выводит:
- первую строку: shape
X, shapeW, shapeYчерез пробел в виде(M, K) (K, N) (M, N); - затем
Mстрок — строки матрицыY, значения через пробел, округлённые до 2 знаков после запятой.
Ввод и вывод
Программа читает stdin и пишет в stdout.
Формат ввода
M K N
X[0][0] ... X[0][K-1]
...
X[M-1][0] ... X[M-1][K-1]
W[0][0] ... W[0][N-1]
...
W[K-1][0] ... W[K-1][N-1]
Гарантируется M >= 2, K >= 2, N >= 2.
Формат вывода
(M, K) (K, N) (M, N)
Y[0][0] ... Y[0][N-1]
...
Y[M-1][0] ... Y[M-1][N-1]
Каждое значение Y выводится с ровно двумя знаками после запятой.
Примеры
Пример 1
- Ввод
2 3 2 1.0 0.0 2.0 0.0 1.0 3.0 1.0 0.0 0.0 1.0 1.0 1.0
- Ожидаемый результат
(2, 3) (3, 2) (2, 2) 3.00 2.00 3.00 4.00
Попробуйте решить
РешениеPython
Без регистрации · результат не сохраняется
