| Safe Haskell | Safe-Inferred |
|---|---|
| Language | Haskell2010 |
Torch.GraduallyTyped.Tensor.Other
Synopsis
- triu :: forall gradient layout device dataType shape. Int -> Tensor gradient layout device dataType shape -> Tensor gradient layout device dataType shape
- tril :: forall gradient layout device dataType shape. Int -> Tensor gradient layout device dataType shape -> Tensor gradient layout device dataType shape
- maskedFill :: forall gradient layout device dataType shape value gradient' layout' device' dataType' shape' shape'' m. (Scalar value, MonadThrow m, Catch (gradient <+> 'Gradient 'WithoutGradient), Catch (dataType <+> 'DataType 'Bool), shape'' ~ BroadcastShapesF shape shape', Catch shape'') => Tensor gradient layout device dataType shape -> value -> Tensor gradient' layout' device' dataType' shape' -> m (Tensor gradient' (layout <+> (layout' <+> 'Layout 'Dense)) (device <+> device') dataType' shape'')
Documentation
Arguments
| :: forall gradient layout device dataType shape. Int | diagonal |
| -> Tensor gradient layout device dataType shape | input |
| -> Tensor gradient layout device dataType shape | output |
triu
Arguments
| :: forall gradient layout device dataType shape. Int | diagonal |
| -> Tensor gradient layout device dataType shape | input |
| -> Tensor gradient layout device dataType shape | output |
tril
Arguments
| :: forall gradient layout device dataType shape value gradient' layout' device' dataType' shape' shape'' m. (Scalar value, MonadThrow m, Catch (gradient <+> 'Gradient 'WithoutGradient), Catch (dataType <+> 'DataType 'Bool), shape'' ~ BroadcastShapesF shape shape', Catch shape'') | |
| => Tensor gradient layout device dataType shape | mask |
| -> value | value |
| -> Tensor gradient' layout' device' dataType' shape' | input |
| -> m (Tensor gradient' (layout <+> (layout' <+> 'Layout 'Dense)) (device <+> device') dataType' shape'') | output |
masked fill