assign¶

dragon.assign(
  inputs,
  starts=None,
  sizes=None,
  copy=False,
  **kwargs
)[source]¶

Assign the value to input.

\[\text{input}[\text{start}:\text{start} + \text{size}, ...] = \text{value} \]

Parameters:

• inputs (Sequence[dragon.Tensor]) – The input and value tensor.
• starts (Union[Sequence[int], dragon.Tensor]], optional) – The start location for each dimension.
• sizes (Union[Sequence[int], dragon.Tensor]], optional) – The number of elements from start.
• copy (bool, optional, default=False) – Return a new tensor or call in-place.

Returns:

dragon.Tensor – The output tensor.