Python: создание numpy по умолчанию для float32

Есть ли какой-либо чистый способ установки numpy для использования значений float32 вместо float64 в глобальном масштабе?

Ответы

Ответ 1

Не то, что я знаю. Вам нужно либо явно указать dtype при вызове конструктора для любого массива, либо передать массив float32 (используйте метод ndarray.astype), прежде чем передавать его в свой код графического процессора (я полагаю, это вопрос, к которому относится этот вопрос).). Если это случай GPU, вас действительно волнует, я одобряю последнее - может стать очень раздражающим, чтобы попытаться сохранить все в единой точности без глубокого понимания правил вещания numpy и очень тщательно разработанного кода.

Другой альтернативой может быть создание собственных методов, которые перегружают стандартные конструкторы numpy (так что numpy.zeros, numpy.ones, numpy.empty). Это должно быть довольно близко к тому, чтобы хранить все в float32.