Слияние кода завершено, страница обновится автоматически
Существующий ге-оператор подключается к aclnnGeScalar, что позволяет избежать преобразования типа при использовании other как скалара, тем самым повышая производительность оператора.
ge
| Номер | Задача | Подзадача | Статус (новый/модифицированный/без изменений/не применимо) | Примечание |
|---|---|---|---|---|
| 1 | Базовая функциональность интерфейса | Primitive | Новый | Добавлен GreaterequalScalar |
| functional | Модифицированный | Переопределение greater_equal_scalar_op.yaml и greater_equal_op.yaml | ||
| tensor | Модифицированный | Переопределение greater_equal_scalar_op.yaml и greater_equal_op.yaml, CPU/GPU и статический граф проходят через py_method процесс | ||
| 2 | Поддержка backend и типов данных | Ascend | Без изменений | |
| GPU | Без изменений | |||
| CPU | Без изменений | |||
| 3 | Поддержка vmap | Без изменений | ||
| 4 | Поддержка динамического Shape | Динамическое Shape | Без изменений | |
| Динамическая Rank | Без изменений | |||
| 5 | Поддержка обратного распространения | bprop функция | Без изменений | |
| Поддержка комплексных чисел | Без изменений | |||
| 6 | Улучшение документации | API отображение | Без изменений | |
| Документация интерфейсов на английском и китайском | Модифицированный | Модификация функциональной документации и документации тензоров на английском и китайском |
1. Сравнение с аналогами
Ссылка на аналогичный интерфейс:
torch.ge
Поддерживаемые типы данных аналога: INT8, UINT8, INT16, INT32, INT64, FLOAT16, FLOAT32, FLOAT64, BFLOAT16, BOOL
2. Ситуация с операторами в MindSpore
Ascend: INT8, UINT8, INT16, UINT16, INT32, UINT32, INT64, UINT64, FLOAT16, FLOAT, DOUBLE, BFLOAT16 (только для 910b), BOOL
INT8, UINT8, INT16, UINT16, INT32, UINT32, INT64, UINT64, FLOAT16, FLOAT, DOUBLE, BFLOAT16 (только для 910b), BOOL
1. Описание функциональности
Проверяет значения входных параметров input >= other поэлементно, выводит Tensor типа bool.
2. Описание интерфейса (mindspore/tensor/nn)
greater_equal:
- op_yaml: greater_equal_scalar_op.yaml
py_method: tensor_greater_equal
Ascend: pyboost
CPU: py_method
GPU: py_method
interface: tensor, function
- op_yaml: greater_equal_op.yaml
py_method: tensor_greater_equal
Ascend: pyboost
CPU: pyboost
GPU: pyboost
interface: tensor, function
- op_yaml: deprecated/greater_equal_method.yaml
py_method: deprecated_tensor_greater_equal
Ascend: py_method
CPU: py_method
GPU: py_method
interface: tensor
x.greater_equal(other)
ge:
alias: greater_equal
x.ge(other)
class GreaterEqualScalar(Primitive):
| Параметр | Тип | Вход/Выход/Атрибут | Описание |
|---|---|---|---|
| input | Tensor | Вход | |
| other | scalar | Вход | |
| output | Tensor | Выход |
Вход Перед тем как оставить комментарий