Functions | |
| Node | einsum (List[Node] inputs, str equation) |
| Return a node which performs Einsum operation. More... | |
| Node | gelu (Node data, str approximation_mode, Optional[str] name=None) |
| Return a node which performs Gelu activation function. More... | |
| Node | roll (NodeInput data, NodeInput shift, NodeInput axes) |
| Return a node which performs Roll operation. More... | |
| Node | gather (NodeInput data, NodeInput indices, NodeInput axis, Optional[int] batch_dims=0) |
| Return a node which performs Gather. More... | |
| Node | dft (NodeInput data, NodeInput axes, Optional[NodeInput] signal_size=None) |
| Return a node which performs DFT operation. More... | |
| Node | idft (NodeInput data, NodeInput axes, Optional[NodeInput] signal_size=None) |
| Return a node which performs IDFT operation. More... | |
| Node ngraph.opset7.ops.dft | ( | NodeInput | data, |
| NodeInput | axes, | ||
| Optional[NodeInput] | signal_size = None |
||
| ) |
Return a node which performs DFT operation.
@param data: Tensor with transformed data. @param axes: Tensor with axes to transform. @param signal_size: Tensor specifying signal size with respect to axes from the input 'axes'. @return: The new node which performs DFT operation on the input data tensor.
| Node ngraph.opset7.ops.einsum | ( | List[Node] | inputs, |
| str | equation | ||
| ) |
Return a node which performs Einsum operation.
@param inputs: The list of input nodes @param equation: Einsum equation @return: The new node performing Einsum operation on the inputs
| Node ngraph.opset7.ops.gather | ( | NodeInput | data, |
| NodeInput | indices, | ||
| NodeInput | axis, | ||
| Optional[int] | batch_dims = 0 |
||
| ) |
Return a node which performs Gather.
@param data: N-D tensor with data for gathering @param indices: N-D tensor with indices by which data is gathered @param axis: axis along which elements are gathered @param batch_dims: number of batch dimensions @return: The new node which performs Gather
| Node ngraph.opset7.ops.gelu | ( | Node | data, |
| str | approximation_mode, | ||
| Optional[str] | name = None |
||
| ) |
Return a node which performs Gelu activation function.
@param data: The node with data tensor.
@param approximation_mode: defines which approximation to use ('tanh' or 'erf')
@param name: Optional output node name.
@return The new node performing a Gelu activation with the input tensor.
| Node ngraph.opset7.ops.idft | ( | NodeInput | data, |
| NodeInput | axes, | ||
| Optional[NodeInput] | signal_size = None |
||
| ) |
Return a node which performs IDFT operation.
@param data: Tensor with transformed data. @param axes: Tensor with axes to transform. @param signal_size: Tensor specifying signal size with respect to axes from the input 'axes'. @return: The new node which performs IDFT operation on the input data tensor.
| Node ngraph.opset7.ops.roll | ( | NodeInput | data, |
| NodeInput | shift, | ||
| NodeInput | axes | ||
| ) |
Return a node which performs Roll operation.
@param data: The node with data tensor. @param shift: The node with the tensor with numbers of places by which elements are shifted. @param axes: The node with the tensor with axes along which elements are shifted. @return The new node performing a Roll operation on the input tensor.