diag
=None) diag(func, size
Creates a diagonal tensor with func on its diagonal.
Parameters
Name | Type | Description | Default |
---|---|---|---|
func | Differentiable or scalar | Symbolic object to set the diagonal to | required |
size | size of the diagonal matrix (size x size). Defaults to the number of spatial dimensions when unspecified | None |