Сохранить как тип массива при возврате скаляра

Массив a = np.array(1) удобно, так как позволяет a.shape **

Как я могу сохранить numpy тип класса при выполнении операции, которая возвращает скаляр, например

b = 0.

c = np.dot(a.T, a)

b+=c

b.shape
---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-19-82ed078d10e8> in <module>()
----> 1 b.shape

AttributeError: 'float' object has no attribute 'shape'

** действительно полезно при выполнении проверок размерности во время сложных операций, которые должны возвращать скаляры (например, многоиндексное тензорное суммирование)

0 ответов

Другие вопросы по тегам