ones_¶

dragon.vm.torch.nn.init.ones_(tensor)[source]¶

Fill tensor with ones.

$$\text{tensor} \leftarrow 1$$

Parameters:

• tensor (dragon.vm.torch.Tensor) – The input tensor.

Returns:

dragon.vm.torch.Tensor – The output tensor.