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.