ones¶
dragon.
ones
(
shape,
dtype='float32',
**kwargs
)[source]¶Return a tensor filled with ones.
\[\text{out} \leftarrow 1 \]x = dragon.ones(shape=(2, 3), dtype='float32')
- Parameters:
- shape (Sequence[Union[int, dragon.Tensor]]) – The tensor shape.
- dtype (str, optional, default='float32') – The optional data type.
- Returns:
dragon.Tensor – The output tensor.