-- generated by using spec/Declarations.yaml
{-# LANGUAGE DataKinds #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE PolyKinds #-}
{-# LANGUAGE QuasiQuotes #-}
{-# LANGUAGE ScopedTypeVariables #-}
{-# LANGUAGE TemplateHaskell #-}

module Torch.Functional.Internal where

import Foreign.ForeignPtr
import System.IO.Unsafe
import Torch.Cast
import Torch.DType
import Torch.Dimname
import Torch.Internal.Cast
import qualified Torch.Internal.Const as ATen
import qualified Torch.Internal.Managed.Cast
import qualified Torch.Internal.Managed.Native as ATen
import qualified Torch.Internal.Managed.Type.Scalar as ATen
import qualified Torch.Internal.Managed.Type.Tensor as ATen
import qualified Torch.Internal.Managed.Type.Tuple as ATen
import qualified Torch.Internal.Type as ATen
import Torch.Scalar
import Torch.Tensor

align_tensors ::
  -- | tensors
  [Tensor] ->
  [Tensor]
align_tensors :: [Tensor] -> [Tensor]
align_tensors [Tensor]
_tensors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr TensorList -> IO (ForeignPtr TensorList)
ATen.align_tensors_l) [Tensor]
_tensors

native_dropout ::
  -- | input
  Tensor ->
  -- | p
  Double ->
  -- | train
  Bool ->
  (Tensor, Tensor)
native_dropout :: Tensor -> Double -> Bool -> (Tensor, Tensor)
native_dropout Tensor
_input Double
_p Bool
_train = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> CDouble -> CBool -> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.native_dropout_tdb) Tensor
_input Double
_p Bool
_train

dropout ::
  -- | input
  Tensor ->
  -- | p
  Double ->
  -- | train
  Bool ->
  Tensor
dropout :: Tensor -> Double -> Bool -> Tensor
dropout Tensor
_input Double
_p Bool
_train = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> CDouble -> CBool -> IO (ForeignPtr Tensor)
ATen.dropout_tdb) Tensor
_input Double
_p Bool
_train

feature_dropout ::
  -- | input
  Tensor ->
  -- | p
  Double ->
  -- | train
  Bool ->
  Tensor
feature_dropout :: Tensor -> Double -> Bool -> Tensor
feature_dropout Tensor
_input Double
_p Bool
_train = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> CDouble -> CBool -> IO (ForeignPtr Tensor)
ATen.feature_dropout_tdb) Tensor
_input Double
_p Bool
_train

alpha_dropout ::
  -- | input
  Tensor ->
  -- | p
  Double ->
  -- | train
  Bool ->
  Tensor
alpha_dropout :: Tensor -> Double -> Bool -> Tensor
alpha_dropout Tensor
_input Double
_p Bool
_train = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> CDouble -> CBool -> IO (ForeignPtr Tensor)
ATen.alpha_dropout_tdb) Tensor
_input Double
_p Bool
_train

feature_alpha_dropout ::
  -- | input
  Tensor ->
  -- | p
  Double ->
  -- | train
  Bool ->
  Tensor
feature_alpha_dropout :: Tensor -> Double -> Bool -> Tensor
feature_alpha_dropout Tensor
_input Double
_p Bool
_train = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> CDouble -> CBool -> IO (ForeignPtr Tensor)
ATen.feature_alpha_dropout_tdb) Tensor
_input Double
_p Bool
_train

abs ::
  -- | self
  Tensor ->
  Tensor
abs :: Tensor -> Tensor
abs Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.abs_t) Tensor
_self

absolute ::
  -- | self
  Tensor ->
  Tensor
absolute :: Tensor -> Tensor
absolute Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.absolute_t) Tensor
_self

angle ::
  -- | self
  Tensor ->
  Tensor
angle :: Tensor -> Tensor
angle Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.angle_t) Tensor
_self

view_as_real ::
  -- | self
  Tensor ->
  Tensor
view_as_real :: Tensor -> Tensor
view_as_real Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.view_as_real_t) Tensor
_self

view_as_complex ::
  -- | self
  Tensor ->
  Tensor
view_as_complex :: Tensor -> Tensor
view_as_complex Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.view_as_complex_t) Tensor
_self

sgn ::
  -- | self
  Tensor ->
  Tensor
sgn :: Tensor -> Tensor
sgn Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.sgn_t) Tensor
_self

real ::
  -- | self
  Tensor ->
  Tensor
real :: Tensor -> Tensor
real Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.real_t) Tensor
_self

imag ::
  -- | self
  Tensor ->
  Tensor
imag :: Tensor -> Tensor
imag Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.imag_t) Tensor
_self

conj ::
  -- | self
  Tensor ->
  Tensor
conj :: Tensor -> Tensor
conj Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.conj_t) Tensor
_self

conj_physical ::
  -- | self
  Tensor ->
  Tensor
conj_physical :: Tensor -> Tensor
conj_physical Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.conj_physical_t) Tensor
_self

resolve_conj ::
  -- | self
  Tensor ->
  Tensor
resolve_conj :: Tensor -> Tensor
resolve_conj Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.resolve_conj_t) Tensor
_self

resolve_neg ::
  -- | self
  Tensor ->
  Tensor
resolve_neg :: Tensor -> Tensor
resolve_neg Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.resolve_neg_t) Tensor
_self

acos ::
  -- | self
  Tensor ->
  Tensor
acos :: Tensor -> Tensor
acos Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.acos_t) Tensor
_self

arccos ::
  -- | self
  Tensor ->
  Tensor
arccos :: Tensor -> Tensor
arccos Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.arccos_t) Tensor
_self

avg_pool1d ::
  -- | self
  Tensor ->
  -- | kernel_size
  Int ->
  -- | stride
  Int ->
  -- | padding
  Int ->
  -- | ceil_mode
  Bool ->
  -- | count_include_pad
  Bool ->
  Tensor
avg_pool1d :: Tensor -> Int -> Int -> Int -> Bool -> Bool -> Tensor
avg_pool1d Tensor
_self Int
_kernel_size Int
_stride Int
_padding Bool
_ceil_mode Bool
_count_include_pad = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> CBool
-> CBool
-> IO (ForeignPtr Tensor)
ATen.avg_pool1d_tlllbb) Tensor
_self Int
_kernel_size Int
_stride Int
_padding Bool
_ceil_mode Bool
_count_include_pad

adaptive_avg_pool1d ::
  -- | self
  Tensor ->
  -- | output_size
  Int ->
  Tensor
adaptive_avg_pool1d :: Tensor -> Int -> Tensor
adaptive_avg_pool1d Tensor
_self Int
_output_size = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr IntArray -> IO (ForeignPtr Tensor)
ATen.adaptive_avg_pool1d_tl) Tensor
_self Int
_output_size

adaptive_max_pool1d ::
  -- | self
  Tensor ->
  -- | output_size
  Int ->
  (Tensor, Tensor)
adaptive_max_pool1d :: Tensor -> Int -> (Tensor, Tensor)
adaptive_max_pool1d Tensor
_self Int
_output_size = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor
-> ForeignPtr IntArray
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.adaptive_max_pool1d_tl) Tensor
_self Int
_output_size

add ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  -- | alpha
  Float ->
  Tensor
add :: Tensor -> Tensor -> Float -> Tensor
add Tensor
_self Tensor
_other Float
_alpha = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.add_tts) Tensor
_self Tensor
_other Float
_alpha

addScalar ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  -- | alpha
  Float ->
  Tensor
addScalar :: Tensor -> Float -> Float -> Tensor
addScalar Tensor
_self Float
_other Float
_alpha = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Scalar -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.add_tss) Tensor
_self Float
_other Float
_alpha

addmv ::
  -- | self
  Tensor ->
  -- | mat
  Tensor ->
  -- | vec
  Tensor ->
  -- | beta
  Float ->
  -- | alpha
  Float ->
  Tensor
addmv :: Tensor -> Tensor -> Tensor -> Float -> Float -> Tensor
addmv Tensor
_self Tensor
_mat Tensor
_vec Float
_beta Float
_alpha = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr Scalar
-> IO (ForeignPtr Tensor)
ATen.addmv_tttss) Tensor
_self Tensor
_mat Tensor
_vec Float
_beta Float
_alpha

addr ::
  -- | self
  Tensor ->
  -- | vec1
  Tensor ->
  -- | vec2
  Tensor ->
  -- | beta
  Float ->
  -- | alpha
  Float ->
  Tensor
addr :: Tensor -> Tensor -> Tensor -> Float -> Float -> Tensor
addr Tensor
_self Tensor
_vec1 Tensor
_vec2 Float
_beta Float
_alpha = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr Scalar
-> IO (ForeignPtr Tensor)
ATen.addr_tttss) Tensor
_self Tensor
_vec1 Tensor
_vec2 Float
_beta Float
_alpha

affine_grid_generator ::
  -- | theta
  Tensor ->
  -- | size
  [Int] ->
  -- | align_corners
  Bool ->
  Tensor
affine_grid_generator :: Tensor -> [Int] -> Bool -> Tensor
affine_grid_generator Tensor
_theta [Int]
_size Bool
_align_corners = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr IntArray -> CBool -> IO (ForeignPtr Tensor)
ATen.affine_grid_generator_tlb) Tensor
_theta [Int]
_size Bool
_align_corners

allDim ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  Tensor
allDim :: Tensor -> Int -> Bool -> Tensor
allDim Tensor
_self Int
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> Int64 -> CBool -> IO (ForeignPtr Tensor)
ATen.all_tlb) Tensor
_self Int
_dim Bool
_keepdim

allWithDimname ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  -- | keepdim
  Bool ->
  Tensor
allWithDimname :: Tensor -> Dimname -> Bool -> Tensor
allWithDimname Tensor
_self Dimname
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Dimname -> CBool -> IO (ForeignPtr Tensor)
ATen.all_tnb) Tensor
_self Dimname
_dim Bool
_keepdim

allclose ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  -- | rtol
  Double ->
  -- | atol
  Double ->
  -- | equal_nan
  Bool ->
  Bool
allclose :: Tensor -> Tensor -> Double -> Double -> Bool -> Bool
allclose Tensor
_self Tensor
_other Double
_rtol Double
_atol Bool
_equal_nan = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor -> CDouble -> CDouble -> CBool -> IO CBool
ATen.allclose_ttddb) Tensor
_self Tensor
_other Double
_rtol Double
_atol Bool
_equal_nan

anyDim ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  Tensor
anyDim :: Tensor -> Int -> Bool -> Tensor
anyDim Tensor
_self Int
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> Int64 -> CBool -> IO (ForeignPtr Tensor)
ATen.any_tlb) Tensor
_self Int
_dim Bool
_keepdim

anyWithDimname ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  -- | keepdim
  Bool ->
  Tensor
anyWithDimname :: Tensor -> Dimname -> Bool -> Tensor
anyWithDimname Tensor
_self Dimname
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Dimname -> CBool -> IO (ForeignPtr Tensor)
ATen.any_tnb) Tensor
_self Dimname
_dim Bool
_keepdim

argmax ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  Tensor
argmax :: Tensor -> Int -> Bool -> Tensor
argmax Tensor
_self Int
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> Int64 -> CBool -> IO (ForeignPtr Tensor)
ATen.argmax_tlb) Tensor
_self Int
_dim Bool
_keepdim

argmin ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  Tensor
argmin :: Tensor -> Int -> Bool -> Tensor
argmin Tensor
_self Int
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> Int64 -> CBool -> IO (ForeignPtr Tensor)
ATen.argmin_tlb) Tensor
_self Int
_dim Bool
_keepdim

acosh ::
  -- | self
  Tensor ->
  Tensor
acosh :: Tensor -> Tensor
acosh Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.acosh_t) Tensor
_self

arccosh ::
  -- | self
  Tensor ->
  Tensor
arccosh :: Tensor -> Tensor
arccosh Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.arccosh_t) Tensor
_self

asinh ::
  -- | self
  Tensor ->
  Tensor
asinh :: Tensor -> Tensor
asinh Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.asinh_t) Tensor
_self

arcsinh ::
  -- | self
  Tensor ->
  Tensor
arcsinh :: Tensor -> Tensor
arcsinh Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.arcsinh_t) Tensor
_self

atanh ::
  -- | self
  Tensor ->
  Tensor
atanh :: Tensor -> Tensor
atanh Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.atanh_t) Tensor
_self

arctanh ::
  -- | self
  Tensor ->
  Tensor
arctanh :: Tensor -> Tensor
arctanh Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.arctanh_t) Tensor
_self

as_strided ::
  -- | self
  Tensor ->
  -- | size
  [Int] ->
  -- | stride
  [Int] ->
  -- | storage_offset
  Int ->
  Tensor
as_strided :: Tensor -> [Int] -> [Int] -> Int -> Tensor
as_strided Tensor
_self [Int]
_size [Int]
_stride Int
_storage_offset = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> Int64
-> IO (ForeignPtr Tensor)
ATen.as_strided_tlll) Tensor
_self [Int]
_size [Int]
_stride Int
_storage_offset

asin ::
  -- | self
  Tensor ->
  Tensor
asin :: Tensor -> Tensor
asin Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.asin_t) Tensor
_self

arcsin ::
  -- | self
  Tensor ->
  Tensor
arcsin :: Tensor -> Tensor
arcsin Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.arcsin_t) Tensor
_self

atan ::
  -- | self
  Tensor ->
  Tensor
atan :: Tensor -> Tensor
atan Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.atan_t) Tensor
_self

arctan ::
  -- | self
  Tensor ->
  Tensor
arctan :: Tensor -> Tensor
arctan Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.arctan_t) Tensor
_self

atleast_1d_t ::
  -- | self
  Tensor ->
  Tensor
atleast_1d_t :: Tensor -> Tensor
atleast_1d_t Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.atleast_1d_t) Tensor
_self

atleast_1d_l ::
  -- | tensors
  [Tensor] ->
  [Tensor]
atleast_1d_l :: [Tensor] -> [Tensor]
atleast_1d_l [Tensor]
_tensors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr TensorList -> IO (ForeignPtr TensorList)
ATen.atleast_1d_l) [Tensor]
_tensors

atleast_2d_t ::
  -- | self
  Tensor ->
  Tensor
atleast_2d_t :: Tensor -> Tensor
atleast_2d_t Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.atleast_2d_t) Tensor
_self

atleast_2d_l ::
  -- | tensors
  [Tensor] ->
  [Tensor]
atleast_2d_l :: [Tensor] -> [Tensor]
atleast_2d_l [Tensor]
_tensors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr TensorList -> IO (ForeignPtr TensorList)
ATen.atleast_2d_l) [Tensor]
_tensors

atleast_3d_t ::
  -- | self
  Tensor ->
  Tensor
atleast_3d_t :: Tensor -> Tensor
atleast_3d_t Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.atleast_3d_t) Tensor
_self

atleast_3d_l ::
  -- | tensors
  [Tensor] ->
  [Tensor]
atleast_3d_l :: [Tensor] -> [Tensor]
atleast_3d_l [Tensor]
_tensors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr TensorList -> IO (ForeignPtr TensorList)
ATen.atleast_3d_l) [Tensor]
_tensors

baddbmm ::
  -- | self
  Tensor ->
  -- | batch1
  Tensor ->
  -- | batch2
  Tensor ->
  -- | beta
  Float ->
  -- | alpha
  Float ->
  Tensor
baddbmm :: Tensor -> Tensor -> Tensor -> Float -> Float -> Tensor
baddbmm Tensor
_self Tensor
_batch1 Tensor
_batch2 Float
_beta Float
_alpha = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr Scalar
-> IO (ForeignPtr Tensor)
ATen.baddbmm_tttss) Tensor
_self Tensor
_batch1 Tensor
_batch2 Float
_beta Float
_alpha

batch_norm ::
  -- | input
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | running_mean
  Tensor ->
  -- | running_var
  Tensor ->
  -- | training
  Bool ->
  -- | momentum
  Double ->
  -- | eps
  Double ->
  -- | cudnn_enabled
  Bool ->
  Tensor
batch_norm :: Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Bool
-> Double
-> Double
-> Bool
-> Tensor
batch_norm Tensor
_input Tensor
_weight Tensor
_bias Tensor
_running_mean Tensor
_running_var Bool
_training Double
_momentum Double
_eps Bool
_cudnn_enabled = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable y cy) =>
(ca
 -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> cx8 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> x8 -> IO y
cast9 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> CBool
-> CDouble
-> CDouble
-> CBool
-> IO (ForeignPtr Tensor)
ATen.batch_norm_tttttbddb) Tensor
_input Tensor
_weight Tensor
_bias Tensor
_running_mean Tensor
_running_var Bool
_training Double
_momentum Double
_eps Bool
_cudnn_enabled

quantized_batch_norm ::
  -- | input
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | mean
  Tensor ->
  -- | var
  Tensor ->
  -- | eps
  Double ->
  -- | output_scale
  Double ->
  -- | output_zero_point
  Int ->
  Tensor
quantized_batch_norm :: Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Double
-> Double
-> Int
-> Tensor
quantized_batch_norm Tensor
_input Tensor
_weight Tensor
_bias Tensor
_mean Tensor
_var Double
_eps Double
_output_scale Int
_output_zero_point = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> IO y
cast8 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> CDouble
-> CDouble
-> Int64
-> IO (ForeignPtr Tensor)
ATen.quantized_batch_norm_tttttddl) Tensor
_input Tensor
_weight Tensor
_bias Tensor
_mean Tensor
_var Double
_eps Double
_output_scale Int
_output_zero_point

bilinear ::
  -- | input1
  Tensor ->
  -- | input2
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  Tensor
bilinear :: Tensor -> Tensor -> Tensor -> Tensor -> Tensor
bilinear Tensor
_input1 Tensor
_input2 Tensor
_weight Tensor
_bias = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> IO (ForeignPtr Tensor)
ATen.bilinear_tttt) Tensor
_input1 Tensor
_input2 Tensor
_weight Tensor
_bias

binary_cross_entropy ::
  -- | self
  Tensor ->
  -- | target
  Tensor ->
  -- | weight
  Tensor ->
  -- | reduction
  Int ->
  Tensor
binary_cross_entropy :: Tensor -> Tensor -> Tensor -> Int -> Tensor
binary_cross_entropy Tensor
_self Tensor
_target Tensor
_weight Int
_reduction = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> Int64
-> IO (ForeignPtr Tensor)
ATen.binary_cross_entropy_tttl) Tensor
_self Tensor
_target Tensor
_weight Int
_reduction

binary_cross_entropy_with_logits ::
  -- | self
  Tensor ->
  -- | target
  Tensor ->
  -- | weight
  Tensor ->
  -- | pos_weight
  Tensor ->
  -- | reduction
  Int ->
  Tensor
binary_cross_entropy_with_logits :: Tensor -> Tensor -> Tensor -> Tensor -> Int -> Tensor
binary_cross_entropy_with_logits Tensor
_self Tensor
_target Tensor
_weight Tensor
_pos_weight Int
_reduction = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> Int64
-> IO (ForeignPtr Tensor)
ATen.binary_cross_entropy_with_logits_ttttl) Tensor
_self Tensor
_target Tensor
_weight Tensor
_pos_weight Int
_reduction

bincount ::
  -- | self
  Tensor ->
  -- | weights
  Tensor ->
  -- | minlength
  Int ->
  Tensor
bincount :: Tensor -> Tensor -> Int -> Tensor
bincount Tensor
_self Tensor
_weights Int
_minlength = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.bincount_ttl) Tensor
_self Tensor
_weights Int
_minlength

bitwise_not ::
  -- | self
  Tensor ->
  Tensor
bitwise_not :: Tensor -> Tensor
bitwise_not Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.bitwise_not_t) Tensor
_self

copysign_tt ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
copysign_tt :: Tensor -> Tensor -> Tensor
copysign_tt Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.copysign_tt) Tensor
_self Tensor
_other

copysign_ts ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
copysign_ts :: Tensor -> Float -> Tensor
copysign_ts Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.copysign_ts) Tensor
_self Float
_other

logical_not ::
  -- | self
  Tensor ->
  Tensor
logical_not :: Tensor -> Tensor
logical_not Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.logical_not_t) Tensor
_self

logical_xor ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
logical_xor :: Tensor -> Tensor -> Tensor
logical_xor Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.logical_xor_tt) Tensor
_self Tensor
_other

logical_and ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
logical_and :: Tensor -> Tensor -> Tensor
logical_and Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.logical_and_tt) Tensor
_self Tensor
_other

logical_or ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
logical_or :: Tensor -> Tensor -> Tensor
logical_or Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.logical_or_tt) Tensor
_self Tensor
_other

bmm ::
  -- | self
  Tensor ->
  -- | mat2
  Tensor ->
  Tensor
bmm :: Tensor -> Tensor -> Tensor
bmm Tensor
_self Tensor
_mat2 = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.bmm_tt) Tensor
_self Tensor
_mat2

broadcast_tensors ::
  -- | tensors
  [Tensor] ->
  [Tensor]
broadcast_tensors :: [Tensor] -> [Tensor]
broadcast_tensors [Tensor]
_tensors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr TensorList -> IO (ForeignPtr TensorList)
ATen.broadcast_tensors_l) [Tensor]
_tensors

broadcast_to ::
  -- | self
  Tensor ->
  -- | size
  [Int] ->
  Tensor
broadcast_to :: Tensor -> [Int] -> Tensor
broadcast_to Tensor
_self [Int]
_size = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr IntArray -> IO (ForeignPtr Tensor)
ATen.broadcast_to_tl) Tensor
_self [Int]
_size

cat ::
  -- | tensors
  [Tensor] ->
  -- | dim
  Int ->
  Tensor
cat :: [Tensor] -> Int -> Tensor
cat [Tensor]
_tensors Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr TensorList -> Int64 -> IO (ForeignPtr Tensor)
ATen.cat_ll) [Tensor]
_tensors Int
_dim

catWithDimname ::
  -- | tensors
  [Tensor] ->
  -- | dim
  Dimname ->
  Tensor
catWithDimname :: [Tensor] -> Dimname -> Tensor
catWithDimname [Tensor]
_tensors Dimname
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr TensorList
-> ForeignPtr Dimname -> IO (ForeignPtr Tensor)
ATen.cat_ln) [Tensor]
_tensors Dimname
_dim

concat_ll ::
  -- | tensors
  [Tensor] ->
  -- | dim
  Int ->
  Tensor
concat_ll :: [Tensor] -> Int -> Tensor
concat_ll [Tensor]
_tensors Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr TensorList -> Int64 -> IO (ForeignPtr Tensor)
ATen.concat_ll) [Tensor]
_tensors Int
_dim

concat_ln ::
  -- | tensors
  [Tensor] ->
  -- | dim
  Dimname ->
  Tensor
concat_ln :: [Tensor] -> Dimname -> Tensor
concat_ln [Tensor]
_tensors Dimname
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr TensorList
-> ForeignPtr Dimname -> IO (ForeignPtr Tensor)
ATen.concat_ln) [Tensor]
_tensors Dimname
_dim

block_diag ::
  -- | tensors
  [Tensor] ->
  Tensor
block_diag :: [Tensor] -> Tensor
block_diag [Tensor]
_tensors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr TensorList -> IO (ForeignPtr Tensor)
ATen.block_diag_l) [Tensor]
_tensors

ceil ::
  -- | self
  Tensor ->
  Tensor
ceil :: Tensor -> Tensor
ceil Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.ceil_t) Tensor
_self

chain_matmul ::
  -- | matrices
  [Tensor] ->
  Tensor
chain_matmul :: [Tensor] -> Tensor
chain_matmul [Tensor]
_matrices = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr TensorList -> IO (ForeignPtr Tensor)
ATen.chain_matmul_l) [Tensor]
_matrices

unsafe_chunk ::
  -- | self
  Tensor ->
  -- | chunks
  Int ->
  -- | dim
  Int ->
  [Tensor]
unsafe_chunk :: Tensor -> Int -> Int -> [Tensor]
unsafe_chunk Tensor
_self Int
_chunks Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> Int64 -> Int64 -> IO (ForeignPtr TensorList)
ATen.unsafe_chunk_tll) Tensor
_self Int
_chunks Int
_dim

chunk ::
  -- | self
  Tensor ->
  -- | chunks
  Int ->
  -- | dim
  Int ->
  [Tensor]
chunk :: Tensor -> Int -> Int -> [Tensor]
chunk Tensor
_self Int
_chunks Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> Int64 -> Int64 -> IO (ForeignPtr TensorList)
ATen.chunk_tll) Tensor
_self Int
_chunks Int
_dim

tensor_split_tll ::
  -- | self
  Tensor ->
  -- | sections
  Int ->
  -- | dim
  Int ->
  [Tensor]
tensor_split_tll :: Tensor -> Int -> Int -> [Tensor]
tensor_split_tll Tensor
_self Int
_sections Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> Int64 -> Int64 -> IO (ForeignPtr TensorList)
ATen.tensor_split_tll) Tensor
_self Int
_sections Int
_dim

-- tensor_split_tll
--   :: Tensor -- ^ self
--   -> [Int] -- ^ indices
--   -> Int -- ^ dim
--   -> [Tensor]
-- tensor_split_tll _self _indices _dim = unsafePerformIO $ (cast3 ATen.tensor_split_tll) _self _indices _dim

-- tensor_split_ttl
--   :: Tensor -- ^ self
--   -> Tensor -- ^ tensor_indices_or_sections
--   -> Int -- ^ dim
--   -> [Tensor]
-- tensor_split_ttl _self _tensor_indices_or_sections _dim = unsafePerformIO $ (cast3 ATen.tensor_split_ttl) _self _tensor_indices_or_sections _dim

clamp_tss ::
  -- | self
  Tensor ->
  -- | min
  Float ->
  -- | max
  Float ->
  Tensor
clamp_tss :: Tensor -> Float -> Float -> Tensor
clamp_tss Tensor
_self Float
_min Float
_max = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Scalar -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.clamp_tss) Tensor
_self Float
_min Float
_max

clamp_ttt ::
  -- | self
  Tensor ->
  -- | min
  Tensor ->
  -- | max
  Tensor ->
  Tensor
clamp_ttt :: Tensor -> Tensor -> Tensor -> Tensor
clamp_ttt Tensor
_self Tensor
_min Tensor
_max = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.clamp_ttt) Tensor
_self Tensor
_min Tensor
_max

clamp_max_ts ::
  -- | self
  Tensor ->
  -- | max
  Float ->
  Tensor
clamp_max_ts :: Tensor -> Float -> Tensor
clamp_max_ts Tensor
_self Float
_max = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.clamp_max_ts) Tensor
_self Float
_max

clamp_max_tt ::
  -- | self
  Tensor ->
  -- | max
  Tensor ->
  Tensor
clamp_max_tt :: Tensor -> Tensor -> Tensor
clamp_max_tt Tensor
_self Tensor
_max = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.clamp_max_tt) Tensor
_self Tensor
_max

clamp_min_ts ::
  -- | self
  Tensor ->
  -- | min
  Float ->
  Tensor
clamp_min_ts :: Tensor -> Float -> Tensor
clamp_min_ts Tensor
_self Float
_min = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.clamp_min_ts) Tensor
_self Float
_min

clamp_min_tt ::
  -- | self
  Tensor ->
  -- | min
  Tensor ->
  Tensor
clamp_min_tt :: Tensor -> Tensor -> Tensor
clamp_min_tt Tensor
_self Tensor
_min = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.clamp_min_tt) Tensor
_self Tensor
_min

clip_tss ::
  -- | self
  Tensor ->
  -- | min
  Float ->
  -- | max
  Float ->
  Tensor
clip_tss :: Tensor -> Float -> Float -> Tensor
clip_tss Tensor
_self Float
_min Float
_max = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Scalar -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.clip_tss) Tensor
_self Float
_min Float
_max

clip_ttt ::
  -- | self
  Tensor ->
  -- | min
  Tensor ->
  -- | max
  Tensor ->
  Tensor
clip_ttt :: Tensor -> Tensor -> Tensor -> Tensor
clip_ttt Tensor
_self Tensor
_min Tensor
_max = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.clip_ttt) Tensor
_self Tensor
_min Tensor
_max

cudnn_is_acceptable ::
  -- | self
  Tensor ->
  Bool
cudnn_is_acceptable :: Tensor -> Bool
cudnn_is_acceptable Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO CBool
ATen.cudnn_is_acceptable_t) Tensor
_self

complex ::
  -- | real
  Tensor ->
  -- | imag
  Tensor ->
  Tensor
complex :: Tensor -> Tensor -> Tensor
complex Tensor
_real Tensor
_imag = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.complex_tt) Tensor
_real Tensor
_imag

polar ::
  -- | abs
  Tensor ->
  -- | angle
  Tensor ->
  Tensor
polar :: Tensor -> Tensor -> Tensor
polar Tensor
_abs Tensor
_angle = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.polar_tt) Tensor
_abs Tensor
_angle

constant_pad_nd ::
  -- | self
  Tensor ->
  -- | pad
  [Int] ->
  -- | value
  Float ->
  Tensor
constant_pad_nd :: Tensor -> [Int] -> Float -> Tensor
constant_pad_nd Tensor
_self [Int]
_pad Float
_value = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr Scalar
-> IO (ForeignPtr Tensor)
ATen.constant_pad_nd_tls) Tensor
_self [Int]
_pad Float
_value

convolution ::
  -- | input
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | stride
  [Int] ->
  -- | padding
  [Int] ->
  -- | dilation
  [Int] ->
  -- | transposed
  Bool ->
  -- | output_padding
  [Int] ->
  -- | groups
  Int ->
  Tensor
convolution :: Tensor
-> Tensor
-> Tensor
-> [Int]
-> [Int]
-> [Int]
-> Bool
-> [Int]
-> Int
-> Tensor
convolution Tensor
_input Tensor
_weight Tensor
_bias [Int]
_stride [Int]
_padding [Int]
_dilation Bool
_transposed [Int]
_output_padding Int
_groups = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable y cy) =>
(ca
 -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> cx8 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> x8 -> IO y
cast9 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> CBool
-> ForeignPtr IntArray
-> Int64
-> IO (ForeignPtr Tensor)
ATen.convolution_tttlllbll) Tensor
_input Tensor
_weight Tensor
_bias [Int]
_stride [Int]
_padding [Int]
_dilation Bool
_transposed [Int]
_output_padding Int
_groups

convolution_overrideable ::
  -- | input
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | stride
  [Int] ->
  -- | padding
  [Int] ->
  -- | dilation
  [Int] ->
  -- | transposed
  Bool ->
  -- | output_padding
  [Int] ->
  -- | groups
  Int ->
  Tensor
convolution_overrideable :: Tensor
-> Tensor
-> Tensor
-> [Int]
-> [Int]
-> [Int]
-> Bool
-> [Int]
-> Int
-> Tensor
convolution_overrideable Tensor
_input Tensor
_weight Tensor
_bias [Int]
_stride [Int]
_padding [Int]
_dilation Bool
_transposed [Int]
_output_padding Int
_groups = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable y cy) =>
(ca
 -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> cx8 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> x8 -> IO y
cast9 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> CBool
-> ForeignPtr IntArray
-> Int64
-> IO (ForeignPtr Tensor)
ATen.convolution_overrideable_tttlllbll) Tensor
_input Tensor
_weight Tensor
_bias [Int]
_stride [Int]
_padding [Int]
_dilation Bool
_transposed [Int]
_output_padding Int
_groups

convolution_backward_overrideable ::
  -- | grad_output
  Tensor ->
  -- | input
  Tensor ->
  -- | weight
  Tensor ->
  -- | stride
  [Int] ->
  -- | padding
  [Int] ->
  -- | dilation
  [Int] ->
  -- | transposed
  Bool ->
  -- | output_padding
  [Int] ->
  -- | groups
  Int ->
  -- | output_mask
  (Bool, Bool, Bool) ->
  (Tensor, Tensor, Tensor)
convolution_backward_overrideable :: Tensor
-> Tensor
-> Tensor
-> [Int]
-> [Int]
-> [Int]
-> Bool
-> [Int]
-> Int
-> (Bool, Bool, Bool)
-> (Tensor, Tensor, Tensor)
convolution_backward_overrideable Tensor
_grad_output Tensor
_input Tensor
_weight [Int]
_stride [Int]
_padding [Int]
_dilation Bool
_transposed [Int]
_output_padding Int
_groups (Bool, Bool, Bool)
_output_mask = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       x9 cx9 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable x9 cx9, Castable y cy) =>
(ca
 -> cx1
 -> cx2
 -> cx3
 -> cx4
 -> cx5
 -> cx6
 -> cx7
 -> cx8
 -> cx9
 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> x8 -> x9 -> IO y
cast10 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> CBool
-> ForeignPtr IntArray
-> Int64
-> ForeignPtr (StdArray '(CBool, 3))
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor, Tensor)))
ATen.convolution_backward_overrideable_tttlllblla) Tensor
_grad_output Tensor
_input Tensor
_weight [Int]
_stride [Int]
_padding [Int]
_dilation Bool
_transposed [Int]
_output_padding Int
_groups (Bool, Bool, Bool)
_output_mask

conv1d_tttllll ::
  -- | input
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | stride
  Int ->
  -- | padding
  Int ->
  -- | dilation
  Int ->
  -- | groups
  Int ->
  Tensor
conv1d_tttllll :: Tensor -> Tensor -> Tensor -> Int -> Int -> Int -> Int -> Tensor
conv1d_tttllll Tensor
_input Tensor
_weight Tensor
_bias Int
_stride Int
_padding Int
_dilation Int
_groups = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> IO y
cast7 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> Int64
-> IO (ForeignPtr Tensor)
ATen.conv1d_tttllll) Tensor
_input Tensor
_weight Tensor
_bias Int
_stride Int
_padding Int
_dilation Int
_groups

conv2d ::
  -- | input
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | stride
  (Int, Int) ->
  -- | padding
  (Int, Int) ->
  -- | dilation
  (Int, Int) ->
  -- | groups
  Int ->
  Tensor
conv2d :: Tensor
-> Tensor
-> Tensor
-> (Int, Int)
-> (Int, Int)
-> (Int, Int)
-> Int
-> Tensor
conv2d Tensor
_input Tensor
_weight Tensor
_bias (Int, Int)
_stride (Int, Int)
_padding (Int, Int)
_dilation Int
_groups = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> IO y
cast7 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> Int64
-> IO (ForeignPtr Tensor)
ATen.conv2d_tttllll) Tensor
_input Tensor
_weight Tensor
_bias (Int, Int)
_stride (Int, Int)
_padding (Int, Int)
_dilation Int
_groups

conv3d_tttllll ::
  -- | input
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | stride
  (Int, Int, Int) ->
  -- | padding
  (Int, Int, Int) ->
  -- | dilation
  (Int, Int, Int) ->
  -- | groups
  Int ->
  Tensor
conv3d_tttllll :: Tensor
-> Tensor
-> Tensor
-> (Int, Int, Int)
-> (Int, Int, Int)
-> (Int, Int, Int)
-> Int
-> Tensor
conv3d_tttllll Tensor
_input Tensor
_weight Tensor
_bias (Int, Int, Int)
_stride (Int, Int, Int)
_padding (Int, Int, Int)
_dilation Int
_groups = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> IO y
cast7 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> Int64
-> IO (ForeignPtr Tensor)
ATen.conv3d_tttllll) Tensor
_input Tensor
_weight Tensor
_bias (Int, Int, Int)
_stride (Int, Int, Int)
_padding (Int, Int, Int)
_dilation Int
_groups

conv1d_tttlsll ::
  -- | input
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | stride
  Int ->
  -- | padding
  String ->
  -- | dilation
  Int ->
  -- | groups
  Int ->
  Tensor
conv1d_tttlsll :: Tensor -> Tensor -> Tensor -> Int -> String -> Int -> Int -> Tensor
conv1d_tttlsll Tensor
_input Tensor
_weight Tensor
_bias Int
_stride String
_padding Int
_dilation Int
_groups = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> IO y
cast7 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr StdString
-> ForeignPtr IntArray
-> Int64
-> IO (ForeignPtr Tensor)
ATen.conv1d_tttlsll) Tensor
_input Tensor
_weight Tensor
_bias Int
_stride String
_padding Int
_dilation Int
_groups

conv2d_tttlsll ::
  -- | input
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | stride
  (Int, Int) ->
  -- | padding
  String ->
  -- | dilation
  (Int, Int) ->
  -- | groups
  Int ->
  Tensor
conv2d_tttlsll :: Tensor
-> Tensor
-> Tensor
-> (Int, Int)
-> String
-> (Int, Int)
-> Int
-> Tensor
conv2d_tttlsll Tensor
_input Tensor
_weight Tensor
_bias (Int, Int)
_stride String
_padding (Int, Int)
_dilation Int
_groups = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> IO y
cast7 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr StdString
-> ForeignPtr IntArray
-> Int64
-> IO (ForeignPtr Tensor)
ATen.conv2d_tttlsll) Tensor
_input Tensor
_weight Tensor
_bias (Int, Int)
_stride String
_padding (Int, Int)
_dilation Int
_groups

conv3d_tttlsll ::
  -- | input
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | stride
  (Int, Int, Int) ->
  -- | padding
  String ->
  -- | dilation
  (Int, Int, Int) ->
  -- | groups
  Int ->
  Tensor
conv3d_tttlsll :: Tensor
-> Tensor
-> Tensor
-> (Int, Int, Int)
-> String
-> (Int, Int, Int)
-> Int
-> Tensor
conv3d_tttlsll Tensor
_input Tensor
_weight Tensor
_bias (Int, Int, Int)
_stride String
_padding (Int, Int, Int)
_dilation Int
_groups = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> IO y
cast7 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr StdString
-> ForeignPtr IntArray
-> Int64
-> IO (ForeignPtr Tensor)
ATen.conv3d_tttlsll) Tensor
_input Tensor
_weight Tensor
_bias (Int, Int, Int)
_stride String
_padding (Int, Int, Int)
_dilation Int
_groups

conv_tbc ::
  -- | self
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | pad
  Int ->
  Tensor
conv_tbc :: Tensor -> Tensor -> Tensor -> Int -> Tensor
conv_tbc Tensor
_self Tensor
_weight Tensor
_bias Int
_pad = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> Int64
-> IO (ForeignPtr Tensor)
ATen.conv_tbc_tttl) Tensor
_self Tensor
_weight Tensor
_bias Int
_pad

conv_transpose1d ::
  -- | input
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | stride
  Int ->
  -- | padding
  Int ->
  -- | output_padding
  Int ->
  -- | groups
  Int ->
  -- | dilation
  Int ->
  Tensor
conv_transpose1d :: Tensor
-> Tensor -> Tensor -> Int -> Int -> Int -> Int -> Int -> Tensor
conv_transpose1d Tensor
_input Tensor
_weight Tensor
_bias Int
_stride Int
_padding Int
_output_padding Int
_groups Int
_dilation = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> IO y
cast8 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> Int64
-> ForeignPtr IntArray
-> IO (ForeignPtr Tensor)
ATen.conv_transpose1d_tttlllll) Tensor
_input Tensor
_weight Tensor
_bias Int
_stride Int
_padding Int
_output_padding Int
_groups Int
_dilation

conv_transpose2d ::
  -- | input
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | stride
  (Int, Int) ->
  -- | padding
  (Int, Int) ->
  -- | output_padding
  (Int, Int) ->
  -- | groups
  Int ->
  -- | dilation
  (Int, Int) ->
  Tensor
conv_transpose2d :: Tensor
-> Tensor
-> Tensor
-> (Int, Int)
-> (Int, Int)
-> (Int, Int)
-> Int
-> (Int, Int)
-> Tensor
conv_transpose2d Tensor
_input Tensor
_weight Tensor
_bias (Int, Int)
_stride (Int, Int)
_padding (Int, Int)
_output_padding Int
_groups (Int, Int)
_dilation = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> IO y
cast8 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> Int64
-> ForeignPtr IntArray
-> IO (ForeignPtr Tensor)
ATen.conv_transpose2d_tttlllll) Tensor
_input Tensor
_weight Tensor
_bias (Int, Int)
_stride (Int, Int)
_padding (Int, Int)
_output_padding Int
_groups (Int, Int)
_dilation

conv_transpose3d ::
  -- | input
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | stride
  (Int, Int, Int) ->
  -- | padding
  (Int, Int, Int) ->
  -- | output_padding
  (Int, Int, Int) ->
  -- | groups
  Int ->
  -- | dilation
  (Int, Int, Int) ->
  Tensor
conv_transpose3d :: Tensor
-> Tensor
-> Tensor
-> (Int, Int, Int)
-> (Int, Int, Int)
-> (Int, Int, Int)
-> Int
-> (Int, Int, Int)
-> Tensor
conv_transpose3d Tensor
_input Tensor
_weight Tensor
_bias (Int, Int, Int)
_stride (Int, Int, Int)
_padding (Int, Int, Int)
_output_padding Int
_groups (Int, Int, Int)
_dilation = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> IO y
cast8 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> Int64
-> ForeignPtr IntArray
-> IO (ForeignPtr Tensor)
ATen.conv_transpose3d_tttlllll) Tensor
_input Tensor
_weight Tensor
_bias (Int, Int, Int)
_stride (Int, Int, Int)
_padding (Int, Int, Int)
_output_padding Int
_groups (Int, Int, Int)
_dilation

cos ::
  -- | self
  Tensor ->
  Tensor
cos :: Tensor -> Tensor
cos Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.cos_t) Tensor
_self

cosh ::
  -- | self
  Tensor ->
  Tensor
cosh :: Tensor -> Tensor
cosh Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.cosh_t) Tensor
_self

cosine_embedding_loss ::
  -- | input1
  Tensor ->
  -- | input2
  Tensor ->
  -- | target
  Tensor ->
  -- | margin
  Double ->
  -- | reduction
  Int ->
  Tensor
cosine_embedding_loss :: Tensor -> Tensor -> Tensor -> Double -> Int -> Tensor
cosine_embedding_loss Tensor
_input1 Tensor
_input2 Tensor
_target Double
_margin Int
_reduction = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> CDouble
-> Int64
-> IO (ForeignPtr Tensor)
ATen.cosine_embedding_loss_tttdl) Tensor
_input1 Tensor
_input2 Tensor
_target Double
_margin Int
_reduction

count_nonzero_tl ::
  -- | self
  Tensor ->
  -- | dim
  [Int] ->
  Tensor
count_nonzero_tl :: Tensor -> [Int] -> Tensor
count_nonzero_tl Tensor
_self [Int]
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr IntArray -> IO (ForeignPtr Tensor)
ATen.count_nonzero_tl) Tensor
_self [Int]
_dim

-- count_nonzero_tl
--   :: Tensor -- ^ self
--   -> Int -- ^ dim
--   -> Tensor
-- count_nonzero_tl _self _dim = unsafePerformIO $ (cast2 ATen.count_nonzero_tl) _self _dim

cov ::
  -- | self
  Tensor ->
  -- | correction
  Int ->
  -- | fweights
  Tensor ->
  -- | aweights
  Tensor ->
  Tensor
cov :: Tensor -> Int -> Tensor -> Tensor -> Tensor
cov Tensor
_self Int
_correction Tensor
_fweights Tensor
_aweights = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> IO (ForeignPtr Tensor)
ATen.cov_tltt) Tensor
_self Int
_correction Tensor
_fweights Tensor
_aweights

corrcoef ::
  -- | self
  Tensor ->
  Tensor
corrcoef :: Tensor -> Tensor
corrcoef Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.corrcoef_t) Tensor
_self

cudnn_affine_grid_generator ::
  -- | theta
  Tensor ->
  -- | N
  Int ->
  -- | C
  Int ->
  -- | H
  Int ->
  -- | W
  Int ->
  Tensor
cudnn_affine_grid_generator :: Tensor -> Int -> Int -> Int -> Int -> Tensor
cudnn_affine_grid_generator Tensor
_theta Int
_N Int
_C Int
_H Int
_W = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> Int64 -> Int64 -> Int64 -> Int64 -> IO (ForeignPtr Tensor)
ATen.cudnn_affine_grid_generator_tllll) Tensor
_theta Int
_N Int
_C Int
_H Int
_W

cudnn_batch_norm ::
  -- | input
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | running_mean
  Tensor ->
  -- | running_var
  Tensor ->
  -- | training
  Bool ->
  -- | exponential_average_factor
  Double ->
  -- | epsilon
  Double ->
  (Tensor, Tensor, Tensor, Tensor)
cudnn_batch_norm :: Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Bool
-> Double
-> Double
-> (Tensor, Tensor, Tensor, Tensor)
cudnn_batch_norm Tensor
_input Tensor
_weight Tensor
_bias Tensor
_running_mean Tensor
_running_var Bool
_training Double
_exponential_average_factor Double
_epsilon = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> IO y
cast8 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> CBool
-> CDouble
-> CDouble
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor, Tensor, Tensor)))
ATen.cudnn_batch_norm_tttttbdd) Tensor
_input Tensor
_weight Tensor
_bias Tensor
_running_mean Tensor
_running_var Bool
_training Double
_exponential_average_factor Double
_epsilon

cudnn_convolution ::
  -- | self
  Tensor ->
  -- | weight
  Tensor ->
  -- | padding
  [Int] ->
  -- | stride
  [Int] ->
  -- | dilation
  [Int] ->
  -- | groups
  Int ->
  -- | benchmark
  Bool ->
  -- | deterministic
  Bool ->
  -- | allow_tf32
  Bool ->
  Tensor
cudnn_convolution :: Tensor
-> Tensor
-> [Int]
-> [Int]
-> [Int]
-> Int
-> Bool
-> Bool
-> Bool
-> Tensor
cudnn_convolution Tensor
_self Tensor
_weight [Int]
_padding [Int]
_stride [Int]
_dilation Int
_groups Bool
_benchmark Bool
_deterministic Bool
_allow_tf32 = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable y cy) =>
(ca
 -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> cx8 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> x8 -> IO y
cast9 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> Int64
-> CBool
-> CBool
-> CBool
-> IO (ForeignPtr Tensor)
ATen.cudnn_convolution_ttllllbbb) Tensor
_self Tensor
_weight [Int]
_padding [Int]
_stride [Int]
_dilation Int
_groups Bool
_benchmark Bool
_deterministic Bool
_allow_tf32

cudnn_convolution_transpose ::
  -- | self
  Tensor ->
  -- | weight
  Tensor ->
  -- | padding
  [Int] ->
  -- | output_padding
  [Int] ->
  -- | stride
  [Int] ->
  -- | dilation
  [Int] ->
  -- | groups
  Int ->
  -- | benchmark
  Bool ->
  -- | deterministic
  Bool ->
  -- | allow_tf32
  Bool ->
  Tensor
cudnn_convolution_transpose :: Tensor
-> Tensor
-> [Int]
-> [Int]
-> [Int]
-> [Int]
-> Int
-> Bool
-> Bool
-> Bool
-> Tensor
cudnn_convolution_transpose Tensor
_self Tensor
_weight [Int]
_padding [Int]
_output_padding [Int]
_stride [Int]
_dilation Int
_groups Bool
_benchmark Bool
_deterministic Bool
_allow_tf32 = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       x9 cx9 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable x9 cx9, Castable y cy) =>
(ca
 -> cx1
 -> cx2
 -> cx3
 -> cx4
 -> cx5
 -> cx6
 -> cx7
 -> cx8
 -> cx9
 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> x8 -> x9 -> IO y
cast10 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> Int64
-> CBool
-> CBool
-> CBool
-> IO (ForeignPtr Tensor)
ATen.cudnn_convolution_transpose_ttlllllbbb) Tensor
_self Tensor
_weight [Int]
_padding [Int]
_output_padding [Int]
_stride [Int]
_dilation Int
_groups Bool
_benchmark Bool
_deterministic Bool
_allow_tf32

cudnn_convolution_relu ::
  -- | self
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | stride
  [Int] ->
  -- | padding
  [Int] ->
  -- | dilation
  [Int] ->
  -- | groups
  Int ->
  Tensor
cudnn_convolution_relu :: Tensor
-> Tensor -> Tensor -> [Int] -> [Int] -> [Int] -> Int -> Tensor
cudnn_convolution_relu Tensor
_self Tensor
_weight Tensor
_bias [Int]
_stride [Int]
_padding [Int]
_dilation Int
_groups = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> IO y
cast7 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> Int64
-> IO (ForeignPtr Tensor)
ATen.cudnn_convolution_relu_tttllll) Tensor
_self Tensor
_weight Tensor
_bias [Int]
_stride [Int]
_padding [Int]
_dilation Int
_groups

cudnn_convolution_add_relu ::
  -- | self
  Tensor ->
  -- | weight
  Tensor ->
  -- | z
  Tensor ->
  -- | alpha
  Float ->
  -- | bias
  Tensor ->
  -- | stride
  [Int] ->
  -- | padding
  [Int] ->
  -- | dilation
  [Int] ->
  -- | groups
  Int ->
  Tensor
cudnn_convolution_add_relu :: Tensor
-> Tensor
-> Tensor
-> Float
-> Tensor
-> [Int]
-> [Int]
-> [Int]
-> Int
-> Tensor
cudnn_convolution_add_relu Tensor
_self Tensor
_weight Tensor
_z Float
_alpha Tensor
_bias [Int]
_stride [Int]
_padding [Int]
_dilation Int
_groups = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable y cy) =>
(ca
 -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> cx8 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> x8 -> IO y
cast9 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> Int64
-> IO (ForeignPtr Tensor)
ATen.cudnn_convolution_add_relu_tttstllll) Tensor
_self Tensor
_weight Tensor
_z Float
_alpha Tensor
_bias [Int]
_stride [Int]
_padding [Int]
_dilation Int
_groups

cudnn_grid_sampler ::
  -- | self
  Tensor ->
  -- | grid
  Tensor ->
  Tensor
cudnn_grid_sampler :: Tensor -> Tensor -> Tensor
cudnn_grid_sampler Tensor
_self Tensor
_grid = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.cudnn_grid_sampler_tt) Tensor
_self Tensor
_grid

cummax_tl ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  (Tensor, Tensor)
cummax_tl :: Tensor -> Int -> (Tensor, Tensor)
cummax_tl Tensor
_self Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor
-> Int64 -> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.cummax_tl) Tensor
_self Int
_dim

cummax_tn ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  (Tensor, Tensor)
cummax_tn :: Tensor -> Dimname -> (Tensor, Tensor)
cummax_tn Tensor
_self Dimname
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor
-> ForeignPtr Dimname
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.cummax_tn) Tensor
_self Dimname
_dim

cummin_tl ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  (Tensor, Tensor)
cummin_tl :: Tensor -> Int -> (Tensor, Tensor)
cummin_tl Tensor
_self Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor
-> Int64 -> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.cummin_tl) Tensor
_self Int
_dim

cummin_tn ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  (Tensor, Tensor)
cummin_tn :: Tensor -> Dimname -> (Tensor, Tensor)
cummin_tn Tensor
_self Dimname
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor
-> ForeignPtr Dimname
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.cummin_tn) Tensor
_self Dimname
_dim

cumprod ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | dtype
  DType ->
  Tensor
cumprod :: Tensor -> Int -> DType -> Tensor
cumprod Tensor
_self Int
_dim DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> Int64 -> ScalarType -> IO (ForeignPtr Tensor)
ATen.cumprod_tls) Tensor
_self Int
_dim DType
_dtype

cumprodWithDimname ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  -- | dtype
  DType ->
  Tensor
cumprodWithDimname :: Tensor -> Dimname -> DType -> Tensor
cumprodWithDimname Tensor
_self Dimname
_dim DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Dimname -> ScalarType -> IO (ForeignPtr Tensor)
ATen.cumprod_tns) Tensor
_self Dimname
_dim DType
_dtype

cumsum ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | dtype
  DType ->
  Tensor
cumsum :: Tensor -> Int -> DType -> Tensor
cumsum Tensor
_self Int
_dim DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> Int64 -> ScalarType -> IO (ForeignPtr Tensor)
ATen.cumsum_tls) Tensor
_self Int
_dim DType
_dtype

cumsumWithDimname ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  -- | dtype
  DType ->
  Tensor
cumsumWithDimname :: Tensor -> Dimname -> DType -> Tensor
cumsumWithDimname Tensor
_self Dimname
_dim DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Dimname -> ScalarType -> IO (ForeignPtr Tensor)
ATen.cumsum_tns) Tensor
_self Dimname
_dim DType
_dtype

cumulative_trapezoid_ttl ::
  -- | y
  Tensor ->
  -- | x
  Tensor ->
  -- | dim
  Int ->
  Tensor
cumulative_trapezoid_ttl :: Tensor -> Tensor -> Int -> Tensor
cumulative_trapezoid_ttl Tensor
_y Tensor
_x Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.cumulative_trapezoid_ttl) Tensor
_y Tensor
_x Int
_dim

cumulative_trapezoid_tsl ::
  -- | y
  Tensor ->
  -- | dx
  Float ->
  -- | dim
  Int ->
  Tensor
cumulative_trapezoid_tsl :: Tensor -> Float -> Int -> Tensor
cumulative_trapezoid_tsl Tensor
_y Float
_dx Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Scalar -> Int64 -> IO (ForeignPtr Tensor)
ATen.cumulative_trapezoid_tsl) Tensor
_y Float
_dx Int
_dim

ctcLoss' ::
  -- | log_probs
  Tensor ->
  -- | targets
  Tensor ->
  -- | input_lengths
  [Int] ->
  -- | target_lengths
  [Int] ->
  -- | blank
  Int ->
  -- | reduction
  Int ->
  -- | zero_infinity
  Bool ->
  Tensor
ctcLoss' :: Tensor -> Tensor -> [Int] -> [Int] -> Int -> Int -> Bool -> Tensor
ctcLoss' Tensor
_log_probs Tensor
_targets [Int]
_input_lengths [Int]
_target_lengths Int
_blank Int
_reduction Bool
_zero_infinity = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> IO y
cast7 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> Int64
-> Int64
-> CBool
-> IO (ForeignPtr Tensor)
ATen.ctc_loss_ttllllb) Tensor
_log_probs Tensor
_targets [Int]
_input_lengths [Int]
_target_lengths Int
_blank Int
_reduction Bool
_zero_infinity

ctcLoss ::
  -- | log_probs
  Tensor ->
  -- | targets
  Tensor ->
  -- | input_lengths
  Tensor ->
  -- | target_lengths
  Tensor ->
  -- | blank
  Int ->
  -- | reduction
  Int ->
  -- | zero_infinity
  Bool ->
  Tensor
ctcLoss :: Tensor
-> Tensor -> Tensor -> Tensor -> Int -> Int -> Bool -> Tensor
ctcLoss Tensor
_log_probs Tensor
_targets Tensor
_input_lengths Tensor
_target_lengths Int
_blank Int
_reduction Bool
_zero_infinity = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> IO y
cast7 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> Int64
-> Int64
-> CBool
-> IO (ForeignPtr Tensor)
ATen.ctc_loss_ttttllb) Tensor
_log_probs Tensor
_targets Tensor
_input_lengths Tensor
_target_lengths Int
_blank Int
_reduction Bool
_zero_infinity

diag_embed ::
  -- | self
  Tensor ->
  -- | offset
  Int ->
  -- | dim1
  Int ->
  -- | dim2
  Int ->
  Tensor
diag_embed :: Tensor -> Int -> Int -> Int -> Tensor
diag_embed Tensor
_self Int
_offset Int
_dim1 Int
_dim2 = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64 -> Int64 -> Int64 -> IO (ForeignPtr Tensor)
ATen.diag_embed_tlll) Tensor
_self Int
_offset Int
_dim1 Int
_dim2

diagflat ::
  -- | self
  Tensor ->
  -- | offset
  Int ->
  Tensor
diagflat :: Tensor -> Int -> Tensor
diagflat Tensor
_self Int
_offset = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.diagflat_tl) Tensor
_self Int
_offset

diagonal_tlll ::
  -- | self
  Tensor ->
  -- | offset
  Int ->
  -- | dim1
  Int ->
  -- | dim2
  Int ->
  Tensor
diagonal_tlll :: Tensor -> Int -> Int -> Int -> Tensor
diagonal_tlll Tensor
_self Int
_offset Int
_dim1 Int
_dim2 = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64 -> Int64 -> Int64 -> IO (ForeignPtr Tensor)
ATen.diagonal_tlll) Tensor
_self Int
_offset Int
_dim1 Int
_dim2

linalg_diagonal ::
  -- | A
  Tensor ->
  -- | offset
  Int ->
  -- | dim1
  Int ->
  -- | dim2
  Int ->
  Tensor
linalg_diagonal :: Tensor -> Int -> Int -> Int -> Tensor
linalg_diagonal Tensor
_A Int
_offset Int
_dim1 Int
_dim2 = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64 -> Int64 -> Int64 -> IO (ForeignPtr Tensor)
ATen.linalg_diagonal_tlll) Tensor
_A Int
_offset Int
_dim1 Int
_dim2

diagonal_tnnnl ::
  -- | self
  Tensor ->
  -- | outdim
  Dimname ->
  -- | dim1
  Dimname ->
  -- | dim2
  Dimname ->
  -- | offset
  Int ->
  Tensor
diagonal_tnnnl :: Tensor -> Dimname -> Dimname -> Dimname -> Int -> Tensor
diagonal_tnnnl Tensor
_self Dimname
_outdim Dimname
_dim1 Dimname
_dim2 Int
_offset = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Dimname
-> ForeignPtr Dimname
-> ForeignPtr Dimname
-> Int64
-> IO (ForeignPtr Tensor)
ATen.diagonal_tnnnl) Tensor
_self Dimname
_outdim Dimname
_dim1 Dimname
_dim2 Int
_offset

diff ::
  -- | self
  Tensor ->
  -- | n
  Int ->
  -- | dim
  Int ->
  -- | prepend
  Tensor ->
  -- | append
  Tensor ->
  Tensor
diff :: Tensor -> Int -> Int -> Tensor -> Tensor -> Tensor
diff Tensor
_self Int
_n Int
_dim Tensor
_prepend Tensor
_append = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> Int64
-> Int64
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> IO (ForeignPtr Tensor)
ATen.diff_tlltt) Tensor
_self Int
_n Int
_dim Tensor
_prepend Tensor
_append

gradient_tsll ::
  -- | self
  Tensor ->
  -- | spacing
  Float ->
  -- | dim
  Int ->
  -- | edge_order
  Int ->
  [Tensor]
gradient_tsll :: Tensor -> Float -> Int -> Int -> [Tensor]
gradient_tsll Tensor
_self Float
_spacing Int
_dim Int
_edge_order = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Scalar
-> Int64
-> Int64
-> IO (ForeignPtr TensorList)
ATen.gradient_tsll) Tensor
_self Float
_spacing Int
_dim Int
_edge_order

-- gradient_tsll
--   :: Tensor -- ^ self
--   -> Float -- ^ spacing
--   -> [Int] -- ^ dim
--   -> Int -- ^ edge_order
--   -> [Tensor]
-- gradient_tsll _self _spacing _dim _edge_order = unsafePerformIO $ (cast4 ATen.gradient_tsll) _self _spacing _dim _edge_order

gradient_tll ::
  -- | self
  Tensor ->
  -- | dim
  [Int] ->
  -- | edge_order
  Int ->
  [Tensor]
gradient_tll :: Tensor -> [Int] -> Int -> [Tensor]
gradient_tll Tensor
_self [Int]
_dim Int
_edge_order = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr IntArray -> Int64 -> IO (ForeignPtr TensorList)
ATen.gradient_tll) Tensor
_self [Int]
_dim Int
_edge_order

-- gradient_tAll
--   :: Tensor -- ^ self
--   -> [Scalar] -- ^ spacing
--   -> Int -- ^ dim
--   -> Int -- ^ edge_order
--   -> [Tensor]
-- gradient_tAll _self _spacing _dim _edge_order = unsafePerformIO $ (cast4 ATen.gradient_tAll) _self _spacing _dim _edge_order

-- gradient_tAll
--   :: Tensor -- ^ self
--   -> [Scalar] -- ^ spacing
--   -> [Int] -- ^ dim
--   -> Int -- ^ edge_order
--   -> [Tensor]
-- gradient_tAll _self _spacing _dim _edge_order = unsafePerformIO $ (cast4 ATen.gradient_tAll) _self _spacing _dim _edge_order

-- gradient_tlll
--   :: Tensor -- ^ self
--   -> [Tensor] -- ^ spacing
--   -> Int -- ^ dim
--   -> Int -- ^ edge_order
--   -> [Tensor]
-- gradient_tlll _self _spacing _dim _edge_order = unsafePerformIO $ (cast4 ATen.gradient_tlll) _self _spacing _dim _edge_order

-- gradient_tlll
--   :: Tensor -- ^ self
--   -> [Tensor] -- ^ spacing
--   -> [Int] -- ^ dim
--   -> Int -- ^ edge_order
--   -> [Tensor]
-- gradient_tlll _self _spacing _dim _edge_order = unsafePerformIO $ (cast4 ATen.gradient_tlll) _self _spacing _dim _edge_order

div ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
div :: Tensor -> Tensor -> Tensor
div Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.div_tt) Tensor
_self Tensor
_other

div_tts ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  -- | rounding_mode
  String ->
  Tensor
div_tts :: Tensor -> Tensor -> String -> Tensor
div_tts Tensor
_self Tensor
_other String
_rounding_mode = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr StdString
-> IO (ForeignPtr Tensor)
ATen.div_tts) Tensor
_self Tensor
_other String
_rounding_mode

divScalar ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
divScalar :: Tensor -> Float -> Tensor
divScalar Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.div_ts) Tensor
_self Float
_other

div_tss ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  -- | rounding_mode
  String ->
  Tensor
div_tss :: Tensor -> Float -> String -> Tensor
div_tss Tensor
_self Float
_other String
_rounding_mode = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr StdString
-> IO (ForeignPtr Tensor)
ATen.div_tss) Tensor
_self Float
_other String
_rounding_mode

divide_tt ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
divide_tt :: Tensor -> Tensor -> Tensor
divide_tt Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.divide_tt) Tensor
_self Tensor
_other

divide_ts ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
divide_ts :: Tensor -> Float -> Tensor
divide_ts Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.divide_ts) Tensor
_self Float
_other

divide_tts ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  -- | rounding_mode
  String ->
  Tensor
divide_tts :: Tensor -> Tensor -> String -> Tensor
divide_tts Tensor
_self Tensor
_other String
_rounding_mode = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr StdString
-> IO (ForeignPtr Tensor)
ATen.divide_tts) Tensor
_self Tensor
_other String
_rounding_mode

divide_tss ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  -- | rounding_mode
  String ->
  Tensor
divide_tss :: Tensor -> Float -> String -> Tensor
divide_tss Tensor
_self Float
_other String
_rounding_mode = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr StdString
-> IO (ForeignPtr Tensor)
ATen.divide_tss) Tensor
_self Float
_other String
_rounding_mode

true_divide_tt ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
true_divide_tt :: Tensor -> Tensor -> Tensor
true_divide_tt Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.true_divide_tt) Tensor
_self Tensor
_other

true_divide_ts ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
true_divide_ts :: Tensor -> Float -> Tensor
true_divide_ts Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.true_divide_ts) Tensor
_self Float
_other

dot ::
  -- | self
  Tensor ->
  -- | tensor
  Tensor ->
  Tensor
dot :: Tensor -> Tensor -> Tensor
dot Tensor
_self Tensor
_tensor = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.dot_tt) Tensor
_self Tensor
_tensor

vdot ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
vdot :: Tensor -> Tensor -> Tensor
vdot Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.vdot_tt) Tensor
_self Tensor
_other

einsum ::
  -- | equation
  String ->
  -- | tensors
  [Tensor] ->
  Tensor
einsum :: String -> [Tensor] -> Tensor
einsum String
_equation [Tensor]
_tensors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr StdString
-> ForeignPtr TensorList -> IO (ForeignPtr Tensor)
ATen.einsum_sl) String
_equation [Tensor]
_tensors

embedding ::
  -- | weight
  Tensor ->
  -- | indices
  Tensor ->
  -- | padding_idx
  Int ->
  -- | scale_grad_by_freq
  Bool ->
  -- | sparse
  Bool ->
  Tensor
embedding :: Tensor -> Tensor -> Int -> Bool -> Bool -> Tensor
embedding Tensor
_weight Tensor
_indices Int
_padding_idx Bool
_scale_grad_by_freq Bool
_sparse = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> Int64
-> CBool
-> CBool
-> IO (ForeignPtr Tensor)
ATen.embedding_ttlbb) Tensor
_weight Tensor
_indices Int
_padding_idx Bool
_scale_grad_by_freq Bool
_sparse

row_stack ::
  -- | tensors
  [Tensor] ->
  Tensor
row_stack :: [Tensor] -> Tensor
row_stack [Tensor]
_tensors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr TensorList -> IO (ForeignPtr Tensor)
ATen.row_stack_l) [Tensor]
_tensors

embedding_bag_tttblbtb ::
  -- | weight
  Tensor ->
  -- | indices
  Tensor ->
  -- | offsets
  Tensor ->
  -- | scale_grad_by_freq
  Bool ->
  -- | mode
  Int ->
  -- | sparse
  Bool ->
  -- | per_sample_weights
  Tensor ->
  -- | include_last_offset
  Bool ->
  (Tensor, Tensor, Tensor, Tensor)
embedding_bag_tttblbtb :: Tensor
-> Tensor
-> Tensor
-> Bool
-> Int
-> Bool
-> Tensor
-> Bool
-> (Tensor, Tensor, Tensor, Tensor)
embedding_bag_tttblbtb Tensor
_weight Tensor
_indices Tensor
_offsets Bool
_scale_grad_by_freq Int
_mode Bool
_sparse Tensor
_per_sample_weights Bool
_include_last_offset = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> IO y
cast8 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> CBool
-> Int64
-> CBool
-> ForeignPtr Tensor
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor, Tensor, Tensor)))
ATen.embedding_bag_tttblbtb) Tensor
_weight Tensor
_indices Tensor
_offsets Bool
_scale_grad_by_freq Int
_mode Bool
_sparse Tensor
_per_sample_weights Bool
_include_last_offset

embedding_bag_tttblbtbl ::
  -- | weight
  Tensor ->
  -- | indices
  Tensor ->
  -- | offsets
  Tensor ->
  -- | scale_grad_by_freq
  Bool ->
  -- | mode
  Int ->
  -- | sparse
  Bool ->
  -- | per_sample_weights
  Tensor ->
  -- | include_last_offset
  Bool ->
  -- | padding_idx
  Int ->
  (Tensor, Tensor, Tensor, Tensor)
embedding_bag_tttblbtbl :: Tensor
-> Tensor
-> Tensor
-> Bool
-> Int
-> Bool
-> Tensor
-> Bool
-> Int
-> (Tensor, Tensor, Tensor, Tensor)
embedding_bag_tttblbtbl Tensor
_weight Tensor
_indices Tensor
_offsets Bool
_scale_grad_by_freq Int
_mode Bool
_sparse Tensor
_per_sample_weights Bool
_include_last_offset Int
_padding_idx = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable y cy) =>
(ca
 -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> cx8 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> x8 -> IO y
cast9 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> CBool
-> Int64
-> CBool
-> ForeignPtr Tensor
-> CBool
-> Int64
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor, Tensor, Tensor)))
ATen.embedding_bag_tttblbtbl) Tensor
_weight Tensor
_indices Tensor
_offsets Bool
_scale_grad_by_freq Int
_mode Bool
_sparse Tensor
_per_sample_weights Bool
_include_last_offset Int
_padding_idx

erf ::
  -- | self
  Tensor ->
  Tensor
erf :: Tensor -> Tensor
erf Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.erf_t) Tensor
_self

erfc ::
  -- | self
  Tensor ->
  Tensor
erfc :: Tensor -> Tensor
erfc Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.erfc_t) Tensor
_self

exp ::
  -- | self
  Tensor ->
  Tensor
exp :: Tensor -> Tensor
exp Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.exp_t) Tensor
_self

exp2 ::
  -- | self
  Tensor ->
  Tensor
exp2 :: Tensor -> Tensor
exp2 Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.exp2_t) Tensor
_self

expm1 ::
  -- | self
  Tensor ->
  Tensor
expm1 :: Tensor -> Tensor
expm1 Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.expm1_t) Tensor
_self

flatten ::
  -- | self
  Tensor ->
  -- | start_dim
  Int ->
  -- | end_dim
  Int ->
  Tensor
flatten :: Tensor -> Int -> Int -> Tensor
flatten Tensor
_self Int
_start_dim Int
_end_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> Int64 -> Int64 -> IO (ForeignPtr Tensor)
ATen.flatten_tll) Tensor
_self Int
_start_dim Int
_end_dim

flattenTo ::
  -- | self
  Tensor ->
  -- | start_dim
  Int ->
  -- | end_dim
  Int ->
  -- | out_dim
  Dimname ->
  Tensor
flattenTo :: Tensor -> Int -> Int -> Dimname -> Tensor
flattenTo Tensor
_self Int
_start_dim Int
_end_dim Dimname
_out_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64 -> Int64 -> ForeignPtr Dimname -> IO (ForeignPtr Tensor)
ATen.flatten_tlln) Tensor
_self Int
_start_dim Int
_end_dim Dimname
_out_dim

flattenToWithDimname ::
  -- | self
  Tensor ->
  -- | start_dim
  Dimname ->
  -- | end_dim
  Dimname ->
  -- | out_dim
  Dimname ->
  Tensor
flattenToWithDimname :: Tensor -> Dimname -> Dimname -> Dimname -> Tensor
flattenToWithDimname Tensor
_self Dimname
_start_dim Dimname
_end_dim Dimname
_out_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Dimname
-> ForeignPtr Dimname
-> ForeignPtr Dimname
-> IO (ForeignPtr Tensor)
ATen.flatten_tnnn) Tensor
_self Dimname
_start_dim Dimname
_end_dim Dimname
_out_dim

flattenToWithDimnames ::
  -- | self
  Tensor ->
  -- | dims
  [Dimname] ->
  -- | out_dim
  Dimname ->
  Tensor
flattenToWithDimnames :: Tensor -> [Dimname] -> Dimname -> Tensor
flattenToWithDimnames Tensor
_self [Dimname]
_dims Dimname
_out_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr DimnameList
-> ForeignPtr Dimname
-> IO (ForeignPtr Tensor)
ATen.flatten_tNn) Tensor
_self [Dimname]
_dims Dimname
_out_dim

floor ::
  -- | self
  Tensor ->
  Tensor
floor :: Tensor -> Tensor
floor Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.floor_t) Tensor
_self

floor_divide_tt ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
floor_divide_tt :: Tensor -> Tensor -> Tensor
floor_divide_tt Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.floor_divide_tt) Tensor
_self Tensor
_other

floor_divide_ts ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
floor_divide_ts :: Tensor -> Float -> Tensor
floor_divide_ts Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.floor_divide_ts) Tensor
_self Float
_other

frac ::
  -- | self
  Tensor ->
  Tensor
frac :: Tensor -> Tensor
frac Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.frac_t) Tensor
_self

gcd ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
gcd :: Tensor -> Tensor -> Tensor
gcd Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.gcd_tt) Tensor
_self Tensor
_other

lcm ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
lcm :: Tensor -> Tensor -> Tensor
lcm Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.lcm_tt) Tensor
_self Tensor
_other

grid_sampler ::
  -- | input
  Tensor ->
  -- | grid
  Tensor ->
  -- | interpolation_mode
  Int ->
  -- | padding_mode
  Int ->
  -- | align_corners
  Bool ->
  Tensor
grid_sampler :: Tensor -> Tensor -> Int -> Int -> Bool -> Tensor
grid_sampler Tensor
_input Tensor
_grid Int
_interpolation_mode Int
_padding_mode Bool
_align_corners = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> Int64
-> Int64
-> CBool
-> IO (ForeignPtr Tensor)
ATen.grid_sampler_ttllb) Tensor
_input Tensor
_grid Int
_interpolation_mode Int
_padding_mode Bool
_align_corners

grid_sampler_2d ::
  -- | input
  Tensor ->
  -- | grid
  Tensor ->
  -- | interpolation_mode
  Int ->
  -- | padding_mode
  Int ->
  -- | align_corners
  Bool ->
  Tensor
grid_sampler_2d :: Tensor -> Tensor -> Int -> Int -> Bool -> Tensor
grid_sampler_2d Tensor
_input Tensor
_grid Int
_interpolation_mode Int
_padding_mode Bool
_align_corners = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> Int64
-> Int64
-> CBool
-> IO (ForeignPtr Tensor)
ATen.grid_sampler_2d_ttllb) Tensor
_input Tensor
_grid Int
_interpolation_mode Int
_padding_mode Bool
_align_corners

grid_sampler_3d ::
  -- | input
  Tensor ->
  -- | grid
  Tensor ->
  -- | interpolation_mode
  Int ->
  -- | padding_mode
  Int ->
  -- | align_corners
  Bool ->
  Tensor
grid_sampler_3d :: Tensor -> Tensor -> Int -> Int -> Bool -> Tensor
grid_sampler_3d Tensor
_input Tensor
_grid Int
_interpolation_mode Int
_padding_mode Bool
_align_corners = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> Int64
-> Int64
-> CBool
-> IO (ForeignPtr Tensor)
ATen.grid_sampler_3d_ttllb) Tensor
_input Tensor
_grid Int
_interpolation_mode Int
_padding_mode Bool
_align_corners

hinge_embedding_loss ::
  -- | self
  Tensor ->
  -- | target
  Tensor ->
  -- | margin
  Double ->
  -- | reduction
  Int ->
  Tensor
hinge_embedding_loss :: Tensor -> Tensor -> Double -> Int -> Tensor
hinge_embedding_loss Tensor
_self Tensor
_target Double
_margin Int
_reduction = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Tensor -> CDouble -> Int64 -> IO (ForeignPtr Tensor)
ATen.hinge_embedding_loss_ttdl) Tensor
_self Tensor
_target Double
_margin Int
_reduction

group_norm ::
  -- | input
  Tensor ->
  -- | num_groups
  Int ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | eps
  Double ->
  -- | cudnn_enabled
  Bool ->
  Tensor
group_norm :: Tensor -> Int -> Tensor -> Tensor -> Double -> Bool -> Tensor
group_norm Tensor
_input Int
_num_groups Tensor
_weight Tensor
_bias Double
_eps Bool
_cudnn_enabled = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> Int64
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> CDouble
-> CBool
-> IO (ForeignPtr Tensor)
ATen.group_norm_tlttdb) Tensor
_input Int
_num_groups Tensor
_weight Tensor
_bias Double
_eps Bool
_cudnn_enabled

native_group_norm ::
  -- | input
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | N
  Int ->
  -- | C
  Int ->
  -- | HxW
  Int ->
  -- | group
  Int ->
  -- | eps
  Double ->
  (Tensor, Tensor, Tensor)
native_group_norm :: Tensor
-> Tensor
-> Tensor
-> Int
-> Int
-> Int
-> Int
-> Double
-> (Tensor, Tensor, Tensor)
native_group_norm Tensor
_input Tensor
_weight Tensor
_bias Int
_N Int
_C Int
_HxW Int
_group Double
_eps = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> IO y
cast8 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> Int64
-> Int64
-> Int64
-> Int64
-> CDouble
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor, Tensor)))
ATen.native_group_norm_tttlllld) Tensor
_input Tensor
_weight Tensor
_bias Int
_N Int
_C Int
_HxW Int
_group Double
_eps

index ::
  -- | self
  Tensor ->
  -- | indices
  [Tensor] ->
  Tensor
index :: Tensor -> [Tensor] -> Tensor
index Tensor
_self [Tensor]
_indices = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor
-> ForeignPtr (C10List (C10Optional Tensor))
-> IO (ForeignPtr Tensor)
ATen.index_tl) Tensor
_self [Tensor]
_indices

indexCopy ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | index
  Tensor ->
  -- | source
  Tensor ->
  Tensor
indexCopy :: Tensor -> Int -> Tensor -> Tensor -> Tensor
indexCopy Tensor
_self Int
_dim Tensor
_index Tensor
_source = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> IO (ForeignPtr Tensor)
ATen.index_copy_tltt) Tensor
_self Int
_dim Tensor
_index Tensor
_source

indexCopyWithDimname ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  -- | index
  Tensor ->
  -- | source
  Tensor ->
  Tensor
indexCopyWithDimname :: Tensor -> Dimname -> Tensor -> Tensor -> Tensor
indexCopyWithDimname Tensor
_self Dimname
_dim Tensor
_index Tensor
_source = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Dimname
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> IO (ForeignPtr Tensor)
ATen.index_copy_tntt) Tensor
_self Dimname
_dim Tensor
_index Tensor
_source

index_put ::
  -- | self
  Tensor ->
  -- | indices
  [Tensor] ->
  -- | values
  Tensor ->
  -- | accumulate
  Bool ->
  Tensor
index_put :: Tensor -> [Tensor] -> Tensor -> Bool -> Tensor
index_put Tensor
_self [Tensor]
_indices Tensor
_values Bool
_accumulate = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr (C10List (C10Optional Tensor))
-> ForeignPtr Tensor
-> CBool
-> IO (ForeignPtr Tensor)
ATen.index_put_tltb) Tensor
_self [Tensor]
_indices Tensor
_values Bool
_accumulate

instance_norm ::
  -- | input
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | running_mean
  Tensor ->
  -- | running_var
  Tensor ->
  -- | use_input_stats
  Bool ->
  -- | momentum
  Double ->
  -- | eps
  Double ->
  -- | cudnn_enabled
  Bool ->
  Tensor
instance_norm :: Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Bool
-> Double
-> Double
-> Bool
-> Tensor
instance_norm Tensor
_input Tensor
_weight Tensor
_bias Tensor
_running_mean Tensor
_running_var Bool
_use_input_stats Double
_momentum Double
_eps Bool
_cudnn_enabled = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable y cy) =>
(ca
 -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> cx8 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> x8 -> IO y
cast9 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> CBool
-> CDouble
-> CDouble
-> CBool
-> IO (ForeignPtr Tensor)
ATen.instance_norm_tttttbddb) Tensor
_input Tensor
_weight Tensor
_bias Tensor
_running_mean Tensor
_running_var Bool
_use_input_stats Double
_momentum Double
_eps Bool
_cudnn_enabled

inverse ::
  -- | self
  Tensor ->
  Tensor
inverse :: Tensor -> Tensor
inverse Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.inverse_t) Tensor
_self

isclose ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  -- | rtol
  Double ->
  -- | atol
  Double ->
  -- | equal_nan
  Bool ->
  Tensor
isclose :: Tensor -> Tensor -> Double -> Double -> Bool -> Tensor
isclose Tensor
_self Tensor
_other Double
_rtol Double
_atol Bool
_equal_nan = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> CDouble
-> CDouble
-> CBool
-> IO (ForeignPtr Tensor)
ATen.isclose_ttddb) Tensor
_self Tensor
_other Double
_rtol Double
_atol Bool
_equal_nan

isin_ttbb ::
  -- | elements
  Tensor ->
  -- | test_elements
  Tensor ->
  -- | assume_unique
  Bool ->
  -- | invert
  Bool ->
  Tensor
isin_ttbb :: Tensor -> Tensor -> Bool -> Bool -> Tensor
isin_ttbb Tensor
_elements Tensor
_test_elements Bool
_assume_unique Bool
_invert = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Tensor -> CBool -> CBool -> IO (ForeignPtr Tensor)
ATen.isin_ttbb) Tensor
_elements Tensor
_test_elements Bool
_assume_unique Bool
_invert

isin_tsbb ::
  -- | elements
  Tensor ->
  -- | test_element
  Float ->
  -- | assume_unique
  Bool ->
  -- | invert
  Bool ->
  Tensor
isin_tsbb :: Tensor -> Float -> Bool -> Bool -> Tensor
isin_tsbb Tensor
_elements Float
_test_element Bool
_assume_unique Bool
_invert = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Scalar -> CBool -> CBool -> IO (ForeignPtr Tensor)
ATen.isin_tsbb) Tensor
_elements Float
_test_element Bool
_assume_unique Bool
_invert

isin_stbb ::
  -- | element
  Float ->
  -- | test_elements
  Tensor ->
  -- | assume_unique
  Bool ->
  -- | invert
  Bool ->
  Tensor
isin_stbb :: Float -> Tensor -> Bool -> Bool -> Tensor
isin_stbb Float
_element Tensor
_test_elements Bool
_assume_unique Bool
_invert = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Scalar
-> ForeignPtr Tensor -> CBool -> CBool -> IO (ForeignPtr Tensor)
ATen.isin_stbb) Float
_element Tensor
_test_elements Bool
_assume_unique Bool
_invert

isnan ::
  -- | self
  Tensor ->
  Tensor
isnan :: Tensor -> Tensor
isnan Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.isnan_t) Tensor
_self

is_distributed ::
  -- | self
  Tensor ->
  Bool
is_distributed :: Tensor -> Bool
is_distributed Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO CBool
ATen.is_distributed_t) Tensor
_self

is_floating_point ::
  -- | self
  Tensor ->
  Bool
is_floating_point :: Tensor -> Bool
is_floating_point Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO CBool
ATen.is_floating_point_t) Tensor
_self

is_complex ::
  -- | self
  Tensor ->
  Bool
is_complex :: Tensor -> Bool
is_complex Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO CBool
ATen.is_complex_t) Tensor
_self

is_conj ::
  -- | self
  Tensor ->
  Bool
is_conj :: Tensor -> Bool
is_conj Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO CBool
ATen.is_conj_t) Tensor
_self

is_neg ::
  -- | self
  Tensor ->
  Bool
is_neg :: Tensor -> Bool
is_neg Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO CBool
ATen.is_neg_t) Tensor
_self

isreal ::
  -- | self
  Tensor ->
  Tensor
isreal :: Tensor -> Tensor
isreal Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.isreal_t) Tensor
_self

is_nonzero ::
  -- | self
  Tensor ->
  Bool
is_nonzero :: Tensor -> Bool
is_nonzero Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO CBool
ATen.is_nonzero_t) Tensor
_self

is_same_size ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Bool
is_same_size :: Tensor -> Tensor -> Bool
is_same_size Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO CBool
ATen.is_same_size_tt) Tensor
_self Tensor
_other

is_signed ::
  -- | self
  Tensor ->
  Bool
is_signed :: Tensor -> Bool
is_signed Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO CBool
ATen.is_signed_t) Tensor
_self

is_inference ::
  -- | self
  Tensor ->
  Bool
is_inference :: Tensor -> Bool
is_inference Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO CBool
ATen.is_inference_t) Tensor
_self

kl_div ::
  -- | self
  Tensor ->
  -- | target
  Tensor ->
  -- | reduction
  Int ->
  -- | log_target
  Bool ->
  Tensor
kl_div :: Tensor -> Tensor -> Int -> Bool -> Tensor
kl_div Tensor
_self Tensor
_target Int
_reduction Bool
_log_target = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Tensor -> Int64 -> CBool -> IO (ForeignPtr Tensor)
ATen.kl_div_ttlb) Tensor
_self Tensor
_target Int
_reduction Bool
_log_target

kron ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
kron :: Tensor -> Tensor -> Tensor
kron Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.kron_tt) Tensor
_self Tensor
_other

kthvalue ::
  -- | self
  Tensor ->
  -- | k
  Int ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  (Tensor, Tensor)
kthvalue :: Tensor -> Int -> Int -> Bool -> (Tensor, Tensor)
kthvalue Tensor
_self Int
_k Int
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64
-> Int64
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.kthvalue_tllb) Tensor
_self Int
_k Int
_dim Bool
_keepdim

kthvalueWithDimname ::
  -- | self
  Tensor ->
  -- | k
  Int ->
  -- | dim
  Dimname ->
  -- | keepdim
  Bool ->
  (Tensor, Tensor)
kthvalueWithDimname :: Tensor -> Int -> Dimname -> Bool -> (Tensor, Tensor)
kthvalueWithDimname Tensor
_self Int
_k Dimname
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64
-> ForeignPtr Dimname
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.kthvalue_tlnb) Tensor
_self Int
_k Dimname
_dim Bool
_keepdim

layer_norm ::
  -- | input
  Tensor ->
  -- | normalized_shape
  [Int] ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | eps
  Double ->
  -- | cudnn_enable
  Bool ->
  Tensor
layer_norm :: Tensor -> [Int] -> Tensor -> Tensor -> Double -> Bool -> Tensor
layer_norm Tensor
_input [Int]
_normalized_shape Tensor
_weight Tensor
_bias Double
_eps Bool
_cudnn_enable = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> CDouble
-> CBool
-> IO (ForeignPtr Tensor)
ATen.layer_norm_tlttdb) Tensor
_input [Int]
_normalized_shape Tensor
_weight Tensor
_bias Double
_eps Bool
_cudnn_enable

native_layer_norm ::
  -- | input
  Tensor ->
  -- | normalized_shape
  [Int] ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | eps
  Double ->
  (Tensor, Tensor, Tensor)
native_layer_norm :: Tensor
-> [Int] -> Tensor -> Tensor -> Double -> (Tensor, Tensor, Tensor)
native_layer_norm Tensor
_input [Int]
_normalized_shape Tensor
_weight Tensor
_bias Double
_eps = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> CDouble
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor, Tensor)))
ATen.native_layer_norm_tlttd) Tensor
_input [Int]
_normalized_shape Tensor
_weight Tensor
_bias Double
_eps

nan_to_num ::
  -- | self
  Tensor ->
  -- | nan
  Double ->
  -- | posinf
  Double ->
  -- | neginf
  Double ->
  Tensor
nan_to_num :: Tensor -> Double -> Double -> Double -> Tensor
nan_to_num Tensor
_self Double
_nan Double
_posinf Double
_neginf = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> CDouble -> CDouble -> CDouble -> IO (ForeignPtr Tensor)
ATen.nan_to_num_tddd) Tensor
_self Double
_nan Double
_posinf Double
_neginf

linear ::
  -- | input
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  Tensor
linear :: Tensor -> Tensor -> Tensor -> Tensor
linear Tensor
_input Tensor
_weight Tensor
_bias = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.linear_ttt) Tensor
_input Tensor
_weight Tensor
_bias

mkldnn_linear ::
  -- | self
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  Tensor
mkldnn_linear :: Tensor -> Tensor -> Tensor -> Tensor
mkldnn_linear Tensor
_self Tensor
_weight Tensor
_bias = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.mkldnn_linear_ttt) Tensor
_self Tensor
_weight Tensor
_bias

mkldnn_linear_backward_input ::
  -- | input_size
  [Int] ->
  -- | grad_output
  Tensor ->
  -- | weight
  Tensor ->
  Tensor
mkldnn_linear_backward_input :: [Int] -> Tensor -> Tensor -> Tensor
mkldnn_linear_backward_input [Int]
_input_size Tensor
_grad_output Tensor
_weight = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr IntArray
-> ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.mkldnn_linear_backward_input_ltt) [Int]
_input_size Tensor
_grad_output Tensor
_weight

mkldnn_linear_backward_weights ::
  -- | grad_output
  Tensor ->
  -- | input
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias_defined
  Bool ->
  (Tensor, Tensor)
mkldnn_linear_backward_weights :: Tensor -> Tensor -> Tensor -> Bool -> (Tensor, Tensor)
mkldnn_linear_backward_weights Tensor
_grad_output Tensor
_input Tensor
_weight Bool
_bias_defined = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.mkldnn_linear_backward_weights_tttb) Tensor
_grad_output Tensor
_input Tensor
_weight Bool
_bias_defined

fbgemm_linear_int8_weight_fp32_activation ::
  -- | input
  Tensor ->
  -- | weight
  Tensor ->
  -- | packed
  Tensor ->
  -- | col_offsets
  Tensor ->
  -- | weight_scale
  Float ->
  -- | weight_zero_point
  Float ->
  -- | bias
  Tensor ->
  Tensor
fbgemm_linear_int8_weight_fp32_activation :: Tensor
-> Tensor -> Tensor -> Tensor -> Float -> Float -> Tensor -> Tensor
fbgemm_linear_int8_weight_fp32_activation Tensor
_input Tensor
_weight Tensor
_packed Tensor
_col_offsets Float
_weight_scale Float
_weight_zero_point Tensor
_bias = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> IO y
cast7 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr Scalar
-> ForeignPtr Tensor
-> IO (ForeignPtr Tensor)
ATen.fbgemm_linear_int8_weight_fp32_activation_ttttsst) Tensor
_input Tensor
_weight Tensor
_packed Tensor
_col_offsets Float
_weight_scale Float
_weight_zero_point Tensor
_bias

fbgemm_linear_int8_weight ::
  -- | input
  Tensor ->
  -- | weight
  Tensor ->
  -- | packed
  Tensor ->
  -- | col_offsets
  Tensor ->
  -- | weight_scale
  Float ->
  -- | weight_zero_point
  Float ->
  -- | bias
  Tensor ->
  Tensor
fbgemm_linear_int8_weight :: Tensor
-> Tensor -> Tensor -> Tensor -> Float -> Float -> Tensor -> Tensor
fbgemm_linear_int8_weight Tensor
_input Tensor
_weight Tensor
_packed Tensor
_col_offsets Float
_weight_scale Float
_weight_zero_point Tensor
_bias = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> IO y
cast7 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr Scalar
-> ForeignPtr Tensor
-> IO (ForeignPtr Tensor)
ATen.fbgemm_linear_int8_weight_ttttsst) Tensor
_input Tensor
_weight Tensor
_packed Tensor
_col_offsets Float
_weight_scale Float
_weight_zero_point Tensor
_bias

fbgemm_linear_quantize_weight ::
  -- | input
  Tensor ->
  (Tensor, Tensor, Double, Int)
fbgemm_linear_quantize_weight :: Tensor -> (Tensor, Tensor, Double, Int)
fbgemm_linear_quantize_weight Tensor
_input = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor, CDouble, Int64)))
ATen.fbgemm_linear_quantize_weight_t) Tensor
_input

fbgemm_pack_gemm_matrix_fp16 ::
  -- | input
  Tensor ->
  Tensor
fbgemm_pack_gemm_matrix_fp16 :: Tensor -> Tensor
fbgemm_pack_gemm_matrix_fp16 Tensor
_input = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.fbgemm_pack_gemm_matrix_fp16_t) Tensor
_input

fbgemm_linear_fp16_weight_fp32_activation ::
  -- | input
  Tensor ->
  -- | packed_weight
  Tensor ->
  -- | bias
  Tensor ->
  Tensor
fbgemm_linear_fp16_weight_fp32_activation :: Tensor -> Tensor -> Tensor -> Tensor
fbgemm_linear_fp16_weight_fp32_activation Tensor
_input Tensor
_packed_weight Tensor
_bias = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.fbgemm_linear_fp16_weight_fp32_activation_ttt) Tensor
_input Tensor
_packed_weight Tensor
_bias

fbgemm_linear_fp16_weight ::
  -- | input
  Tensor ->
  -- | packed_weight
  Tensor ->
  -- | bias
  Tensor ->
  Tensor
fbgemm_linear_fp16_weight :: Tensor -> Tensor -> Tensor -> Tensor
fbgemm_linear_fp16_weight Tensor
_input Tensor
_packed_weight Tensor
_bias = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.fbgemm_linear_fp16_weight_ttt) Tensor
_input Tensor
_packed_weight Tensor
_bias

quantizeFbgemm' ::
  -- | input
  Tensor ->
  Tensor
quantizeFbgemm' :: Tensor -> Tensor
quantizeFbgemm' Tensor
_input = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.fbgemm_pack_quantized_matrix_t) Tensor
_input

quantizeFbgemm ::
  -- | input
  Tensor ->
  -- | K
  Int ->
  -- | N
  Int ->
  Tensor
quantizeFbgemm :: Tensor -> Int -> Int -> Tensor
quantizeFbgemm Tensor
_input Int
_K Int
_N = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> Int64 -> Int64 -> IO (ForeignPtr Tensor)
ATen.fbgemm_pack_quantized_matrix_tll) Tensor
_input Int
_K Int
_N

ldexp ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
ldexp :: Tensor -> Tensor -> Tensor
ldexp Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.ldexp_tt) Tensor
_self Tensor
_other

log ::
  -- | self
  Tensor ->
  Tensor
log :: Tensor -> Tensor
log Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.log_t) Tensor
_self

log10 ::
  -- | self
  Tensor ->
  Tensor
log10 :: Tensor -> Tensor
log10 Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.log10_t) Tensor
_self

log1p ::
  -- | self
  Tensor ->
  Tensor
log1p :: Tensor -> Tensor
log1p Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.log1p_t) Tensor
_self

log2 ::
  -- | self
  Tensor ->
  Tensor
log2 :: Tensor -> Tensor
log2 Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.log2_t) Tensor
_self

logaddexp ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
logaddexp :: Tensor -> Tensor -> Tensor
logaddexp Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.logaddexp_tt) Tensor
_self Tensor
_other

logaddexp2 ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
logaddexp2 :: Tensor -> Tensor -> Tensor
logaddexp2 Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.logaddexp2_tt) Tensor
_self Tensor
_other

xlogy_tt ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
xlogy_tt :: Tensor -> Tensor -> Tensor
xlogy_tt Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.xlogy_tt) Tensor
_self Tensor
_other

xlogy_st ::
  -- | self
  Float ->
  -- | other
  Tensor ->
  Tensor
xlogy_st :: Float -> Tensor -> Tensor
xlogy_st Float
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Scalar -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.xlogy_st) Float
_self Tensor
_other

xlogy_ts ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
xlogy_ts :: Tensor -> Float -> Tensor
xlogy_ts Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.xlogy_ts) Tensor
_self Float
_other

logdet ::
  -- | self
  Tensor ->
  Tensor
logdet :: Tensor -> Tensor
logdet Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.logdet_t) Tensor
_self

logSoftmax ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | dtype
  DType ->
  Tensor
logSoftmax :: Tensor -> Int -> DType -> Tensor
logSoftmax Tensor
_self Int
_dim DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> Int64 -> ScalarType -> IO (ForeignPtr Tensor)
ATen.log_softmax_tls) Tensor
_self Int
_dim DType
_dtype

logSoftmaxWithDimname ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  -- | dtype
  DType ->
  Tensor
logSoftmaxWithDimname :: Tensor -> Dimname -> DType -> Tensor
logSoftmaxWithDimname Tensor
_self Dimname
_dim DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Dimname -> ScalarType -> IO (ForeignPtr Tensor)
ATen.log_softmax_tns) Tensor
_self Dimname
_dim DType
_dtype

logcumsumexp_tl ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  Tensor
logcumsumexp_tl :: Tensor -> Int -> Tensor
logcumsumexp_tl Tensor
_self Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.logcumsumexp_tl) Tensor
_self Int
_dim

logcumsumexp_tn ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  Tensor
logcumsumexp_tn :: Tensor -> Dimname -> Tensor
logcumsumexp_tn Tensor
_self Dimname
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Dimname -> IO (ForeignPtr Tensor)
ATen.logcumsumexp_tn) Tensor
_self Dimname
_dim

logsumexp ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  Tensor
logsumexp :: Tensor -> Int -> Bool -> Tensor
logsumexp Tensor
_self Int
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr IntArray -> CBool -> IO (ForeignPtr Tensor)
ATen.logsumexp_tlb) Tensor
_self Int
_dim Bool
_keepdim

logsumexpWithDimnameList ::
  -- | self
  Tensor ->
  -- | dim
  [Dimname] ->
  -- | keepdim
  Bool ->
  Tensor
logsumexpWithDimnameList :: Tensor -> [Dimname] -> Bool -> Tensor
logsumexpWithDimnameList Tensor
_self [Dimname]
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr DimnameList -> CBool -> IO (ForeignPtr Tensor)
ATen.logsumexp_tNb) Tensor
_self [Dimname]
_dim Bool
_keepdim

margin_ranking_loss ::
  -- | input1
  Tensor ->
  -- | input2
  Tensor ->
  -- | target
  Tensor ->
  -- | margin
  Double ->
  -- | reduction
  Int ->
  Tensor
margin_ranking_loss :: Tensor -> Tensor -> Tensor -> Double -> Int -> Tensor
margin_ranking_loss Tensor
_input1 Tensor
_input2 Tensor
_target Double
_margin Int
_reduction = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> CDouble
-> Int64
-> IO (ForeignPtr Tensor)
ATen.margin_ranking_loss_tttdl) Tensor
_input1 Tensor
_input2 Tensor
_target Double
_margin Int
_reduction

matmul ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
matmul :: Tensor -> Tensor -> Tensor
matmul Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.matmul_tt) Tensor
_self Tensor
_other

matrixRank ::
  -- | self
  Tensor ->
  -- | tol
  Double ->
  -- | symmetric
  Bool ->
  Tensor
matrixRank :: Tensor -> Double -> Bool -> Tensor
matrixRank Tensor
_self Double
_tol Bool
_symmetric = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> CDouble -> CBool -> IO (ForeignPtr Tensor)
ATen.matrix_rank_tdb) Tensor
_self Double
_tol Bool
_symmetric

matrixRank' ::
  -- | self
  Tensor ->
  -- | symmetric
  Bool ->
  Tensor
matrixRank' :: Tensor -> Bool -> Tensor
matrixRank' Tensor
_self Bool
_symmetric = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> CBool -> IO (ForeignPtr Tensor)
ATen.matrix_rank_tb) Tensor
_self Bool
_symmetric

matrix_power ::
  -- | self
  Tensor ->
  -- | n
  Int ->
  Tensor
matrix_power :: Tensor -> Int -> Tensor
matrix_power Tensor
_self Int
_n = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.matrix_power_tl) Tensor
_self Int
_n

matrix_exp ::
  -- | self
  Tensor ->
  Tensor
matrix_exp :: Tensor -> Tensor
matrix_exp Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.matrix_exp_t) Tensor
_self

aminmax ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  (Tensor, Tensor)
aminmax :: Tensor -> Int -> Bool -> (Tensor, Tensor)
aminmax Tensor
_self Int
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> Int64 -> CBool -> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.aminmax_tlb) Tensor
_self Int
_dim Bool
_keepdim

maxDim ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  (Tensor, Tensor)
maxDim :: Tensor -> Int -> Bool -> (Tensor, Tensor)
maxDim Tensor
_self Int
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> Int64 -> CBool -> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.max_tlb) Tensor
_self Int
_dim Bool
_keepdim

maxWithDimname ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  -- | keepdim
  Bool ->
  (Tensor, Tensor)
maxWithDimname :: Tensor -> Dimname -> Bool -> (Tensor, Tensor)
maxWithDimname Tensor
_self Dimname
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Dimname
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.max_tnb) Tensor
_self Dimname
_dim Bool
_keepdim

amax ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  Tensor
amax :: Tensor -> Int -> Bool -> Tensor
amax Tensor
_self Int
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr IntArray -> CBool -> IO (ForeignPtr Tensor)
ATen.amax_tlb) Tensor
_self Int
_dim Bool
_keepdim

max_pool1d_with_indices ::
  -- | self
  Tensor ->
  -- | kernel_size
  Int ->
  -- | stride
  Int ->
  -- | padding
  Int ->
  -- | dilation
  Int ->
  -- | ceil_mode
  Bool ->
  (Tensor, Tensor)
max_pool1d_with_indices :: Tensor -> Int -> Int -> Int -> Int -> Bool -> (Tensor, Tensor)
max_pool1d_with_indices Tensor
_self Int
_kernel_size Int
_stride Int
_padding Int
_dilation Bool
_ceil_mode = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.max_pool1d_with_indices_tllllb) Tensor
_self Int
_kernel_size Int
_stride Int
_padding Int
_dilation Bool
_ceil_mode

max_pool1d ::
  -- | self
  Tensor ->
  -- | kernel_size
  Int ->
  -- | stride
  Int ->
  -- | padding
  Int ->
  -- | dilation
  Int ->
  -- | ceil_mode
  Bool ->
  Tensor
max_pool1d :: Tensor -> Int -> Int -> Int -> Int -> Bool -> Tensor
max_pool1d Tensor
_self Int
_kernel_size Int
_stride Int
_padding Int
_dilation Bool
_ceil_mode = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> CBool
-> IO (ForeignPtr Tensor)
ATen.max_pool1d_tllllb) Tensor
_self Int
_kernel_size Int
_stride Int
_padding Int
_dilation Bool
_ceil_mode

max_pool2d ::
  -- | self
  Tensor ->
  -- | kernel_size
  (Int, Int) ->
  -- | stride
  (Int, Int) ->
  -- | padding
  (Int, Int) ->
  -- | dilation
  (Int, Int) ->
  -- | ceil_mode
  Bool ->
  Tensor
max_pool2d :: Tensor
-> (Int, Int)
-> (Int, Int)
-> (Int, Int)
-> (Int, Int)
-> Bool
-> Tensor
max_pool2d Tensor
_self (Int, Int)
_kernel_size (Int, Int)
_stride (Int, Int)
_padding (Int, Int)
_dilation Bool
_ceil_mode = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> CBool
-> IO (ForeignPtr Tensor)
ATen.max_pool2d_tllllb) Tensor
_self (Int, Int)
_kernel_size (Int, Int)
_stride (Int, Int)
_padding (Int, Int)
_dilation Bool
_ceil_mode

mkldnn_max_pool2d ::
  -- | self
  Tensor ->
  -- | kernel_size
  (Int, Int) ->
  -- | stride
  (Int, Int) ->
  -- | padding
  (Int, Int) ->
  -- | dilation
  (Int, Int) ->
  -- | ceil_mode
  Bool ->
  Tensor
mkldnn_max_pool2d :: Tensor
-> (Int, Int)
-> (Int, Int)
-> (Int, Int)
-> (Int, Int)
-> Bool
-> Tensor
mkldnn_max_pool2d Tensor
_self (Int, Int)
_kernel_size (Int, Int)
_stride (Int, Int)
_padding (Int, Int)
_dilation Bool
_ceil_mode = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> CBool
-> IO (ForeignPtr Tensor)
ATen.mkldnn_max_pool2d_tllllb) Tensor
_self (Int, Int)
_kernel_size (Int, Int)
_stride (Int, Int)
_padding (Int, Int)
_dilation Bool
_ceil_mode

mkldnn_max_pool3d ::
  -- | self
  Tensor ->
  -- | kernel_size
  (Int, Int, Int) ->
  -- | stride
  (Int, Int, Int) ->
  -- | padding
  (Int, Int, Int) ->
  -- | dilation
  (Int, Int, Int) ->
  -- | ceil_mode
  Bool ->
  Tensor
mkldnn_max_pool3d :: Tensor
-> (Int, Int, Int)
-> (Int, Int, Int)
-> (Int, Int, Int)
-> (Int, Int, Int)
-> Bool
-> Tensor
mkldnn_max_pool3d Tensor
_self (Int, Int, Int)
_kernel_size (Int, Int, Int)
_stride (Int, Int, Int)
_padding (Int, Int, Int)
_dilation Bool
_ceil_mode = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> CBool
-> IO (ForeignPtr Tensor)
ATen.mkldnn_max_pool3d_tllllb) Tensor
_self (Int, Int, Int)
_kernel_size (Int, Int, Int)
_stride (Int, Int, Int)
_padding (Int, Int, Int)
_dilation Bool
_ceil_mode

quantized_max_pool1d ::
  -- | self
  Tensor ->
  -- | kernel_size
  Int ->
  -- | stride
  Int ->
  -- | padding
  Int ->
  -- | dilation
  Int ->
  -- | ceil_mode
  Bool ->
  Tensor
quantized_max_pool1d :: Tensor -> Int -> Int -> Int -> Int -> Bool -> Tensor
quantized_max_pool1d Tensor
_self Int
_kernel_size Int
_stride Int
_padding Int
_dilation Bool
_ceil_mode = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> CBool
-> IO (ForeignPtr Tensor)
ATen.quantized_max_pool1d_tllllb) Tensor
_self Int
_kernel_size Int
_stride Int
_padding Int
_dilation Bool
_ceil_mode

quantized_max_pool2d ::
  -- | self
  Tensor ->
  -- | kernel_size
  (Int, Int) ->
  -- | stride
  (Int, Int) ->
  -- | padding
  (Int, Int) ->
  -- | dilation
  (Int, Int) ->
  -- | ceil_mode
  Bool ->
  Tensor
quantized_max_pool2d :: Tensor
-> (Int, Int)
-> (Int, Int)
-> (Int, Int)
-> (Int, Int)
-> Bool
-> Tensor
quantized_max_pool2d Tensor
_self (Int, Int)
_kernel_size (Int, Int)
_stride (Int, Int)
_padding (Int, Int)
_dilation Bool
_ceil_mode = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> CBool
-> IO (ForeignPtr Tensor)
ATen.quantized_max_pool2d_tllllb) Tensor
_self (Int, Int)
_kernel_size (Int, Int)
_stride (Int, Int)
_padding (Int, Int)
_dilation Bool
_ceil_mode

max_pool3d ::
  -- | self
  Tensor ->
  -- | kernel_size
  (Int, Int, Int) ->
  -- | stride
  (Int, Int, Int) ->
  -- | padding
  (Int, Int, Int) ->
  -- | dilation
  (Int, Int, Int) ->
  -- | ceil_mode
  Bool ->
  Tensor
max_pool3d :: Tensor
-> (Int, Int, Int)
-> (Int, Int, Int)
-> (Int, Int, Int)
-> (Int, Int, Int)
-> Bool
-> Tensor
max_pool3d Tensor
_self (Int, Int, Int)
_kernel_size (Int, Int, Int)
_stride (Int, Int, Int)
_padding (Int, Int, Int)
_dilation Bool
_ceil_mode = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> CBool
-> IO (ForeignPtr Tensor)
ATen.max_pool3d_tllllb) Tensor
_self (Int, Int, Int)
_kernel_size (Int, Int, Int)
_stride (Int, Int, Int)
_padding (Int, Int, Int)
_dilation Bool
_ceil_mode

meanAll ::
  -- | self
  Tensor ->
  -- | dtype
  DType ->
  Tensor
meanAll :: Tensor -> DType -> Tensor
meanAll Tensor
_self DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ScalarType -> IO (ForeignPtr Tensor)
ATen.mean_ts) Tensor
_self DType
_dtype

meanDim ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  -- | dtype
  DType ->
  Tensor
meanDim :: Tensor -> Int -> Bool -> DType -> Tensor
meanDim Tensor
_self Int
_dim Bool
_keepdim DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> CBool
-> ScalarType
-> IO (ForeignPtr Tensor)
ATen.mean_tlbs) Tensor
_self Int
_dim Bool
_keepdim DType
_dtype

meanWithDimnames ::
  -- | self
  Tensor ->
  -- | dim
  [Dimname] ->
  -- | keepdim
  Bool ->
  -- | dtype
  DType ->
  Tensor
meanWithDimnames :: Tensor -> [Dimname] -> Bool -> DType -> Tensor
meanWithDimnames Tensor
_self [Dimname]
_dim Bool
_keepdim DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr DimnameList
-> CBool
-> ScalarType
-> IO (ForeignPtr Tensor)
ATen.mean_tNbs) Tensor
_self [Dimname]
_dim Bool
_keepdim DType
_dtype

nanmean ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  -- | dtype
  DType ->
  Tensor
nanmean :: Tensor -> Int -> Bool -> DType -> Tensor
nanmean Tensor
_self Int
_dim Bool
_keepdim DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> CBool
-> ScalarType
-> IO (ForeignPtr Tensor)
ATen.nanmean_tlbs) Tensor
_self Int
_dim Bool
_keepdim DType
_dtype

medianAll ::
  -- | self
  Tensor ->
  Tensor
medianAll :: Tensor -> Tensor
medianAll Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.median_t) Tensor
_self

medianDim ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  (Tensor, Tensor)
medianDim :: Tensor -> Int -> Bool -> (Tensor, Tensor)
medianDim Tensor
_self Int
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> Int64 -> CBool -> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.median_tlb) Tensor
_self Int
_dim Bool
_keepdim

medianWithDimname ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  -- | keepdim
  Bool ->
  (Tensor, Tensor)
medianWithDimname :: Tensor -> Dimname -> Bool -> (Tensor, Tensor)
medianWithDimname Tensor
_self Dimname
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Dimname
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.median_tnb) Tensor
_self Dimname
_dim Bool
_keepdim

nanmedian_t ::
  -- | self
  Tensor ->
  Tensor
nanmedian_t :: Tensor -> Tensor
nanmedian_t Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.nanmedian_t) Tensor
_self

nanmedian_tlb ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  (Tensor, Tensor)
nanmedian_tlb :: Tensor -> Int -> Bool -> (Tensor, Tensor)
nanmedian_tlb Tensor
_self Int
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> Int64 -> CBool -> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.nanmedian_tlb) Tensor
_self Int
_dim Bool
_keepdim

nanmedian_tnb ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  -- | keepdim
  Bool ->
  (Tensor, Tensor)
nanmedian_tnb :: Tensor -> Dimname -> Bool -> (Tensor, Tensor)
nanmedian_tnb Tensor
_self Dimname
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Dimname
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.nanmedian_tnb) Tensor
_self Dimname
_dim Bool
_keepdim

minDim ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  (Tensor, Tensor)
minDim :: Tensor -> Int -> Bool -> (Tensor, Tensor)
minDim Tensor
_self Int
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> Int64 -> CBool -> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.min_tlb) Tensor
_self Int
_dim Bool
_keepdim

minWithDimname ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  -- | keepdim
  Bool ->
  (Tensor, Tensor)
minWithDimname :: Tensor -> Dimname -> Bool -> (Tensor, Tensor)
minWithDimname Tensor
_self Dimname
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Dimname
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.min_tnb) Tensor
_self Dimname
_dim Bool
_keepdim

amin ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  Tensor
amin :: Tensor -> Int -> Bool -> Tensor
amin Tensor
_self Int
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr IntArray -> CBool -> IO (ForeignPtr Tensor)
ATen.amin_tlb) Tensor
_self Int
_dim Bool
_keepdim

mkldnn_convolution ::
  -- | self
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | padding
  [Int] ->
  -- | stride
  [Int] ->
  -- | dilation
  [Int] ->
  -- | groups
  Int ->
  Tensor
mkldnn_convolution :: Tensor
-> Tensor -> Tensor -> [Int] -> [Int] -> [Int] -> Int -> Tensor
mkldnn_convolution Tensor
_self Tensor
_weight Tensor
_bias [Int]
_padding [Int]
_stride [Int]
_dilation Int
_groups = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> IO y
cast7 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> Int64
-> IO (ForeignPtr Tensor)
ATen.mkldnn_convolution_tttllll) Tensor
_self Tensor
_weight Tensor
_bias [Int]
_padding [Int]
_stride [Int]
_dilation Int
_groups

miopen_batch_norm ::
  -- | input
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | running_mean
  Tensor ->
  -- | running_var
  Tensor ->
  -- | training
  Bool ->
  -- | exponential_average_factor
  Double ->
  -- | epsilon
  Double ->
  (Tensor, Tensor, Tensor)
miopen_batch_norm :: Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Bool
-> Double
-> Double
-> (Tensor, Tensor, Tensor)
miopen_batch_norm Tensor
_input Tensor
_weight Tensor
_bias Tensor
_running_mean Tensor
_running_var Bool
_training Double
_exponential_average_factor Double
_epsilon = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> IO y
cast8 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> CBool
-> CDouble
-> CDouble
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor, Tensor)))
ATen.miopen_batch_norm_tttttbdd) Tensor
_input Tensor
_weight Tensor
_bias Tensor
_running_mean Tensor
_running_var Bool
_training Double
_exponential_average_factor Double
_epsilon

miopen_convolution ::
  -- | self
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | padding
  [Int] ->
  -- | stride
  [Int] ->
  -- | dilation
  [Int] ->
  -- | groups
  Int ->
  -- | benchmark
  Bool ->
  -- | deterministic
  Bool ->
  Tensor
miopen_convolution :: Tensor
-> Tensor
-> Tensor
-> [Int]
-> [Int]
-> [Int]
-> Int
-> Bool
-> Bool
-> Tensor
miopen_convolution Tensor
_self Tensor
_weight Tensor
_bias [Int]
_padding [Int]
_stride [Int]
_dilation Int
_groups Bool
_benchmark Bool
_deterministic = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable y cy) =>
(ca
 -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> cx8 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> x8 -> IO y
cast9 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> Int64
-> CBool
-> CBool
-> IO (ForeignPtr Tensor)
ATen.miopen_convolution_tttllllbb) Tensor
_self Tensor
_weight Tensor
_bias [Int]
_padding [Int]
_stride [Int]
_dilation Int
_groups Bool
_benchmark Bool
_deterministic

miopen_convolution_transpose ::
  -- | self
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | padding
  [Int] ->
  -- | output_padding
  [Int] ->
  -- | stride
  [Int] ->
  -- | dilation
  [Int] ->
  -- | groups
  Int ->
  -- | benchmark
  Bool ->
  -- | deterministic
  Bool ->
  Tensor
miopen_convolution_transpose :: Tensor
-> Tensor
-> Tensor
-> [Int]
-> [Int]
-> [Int]
-> [Int]
-> Int
-> Bool
-> Bool
-> Tensor
miopen_convolution_transpose Tensor
_self Tensor
_weight Tensor
_bias [Int]
_padding [Int]
_output_padding [Int]
_stride [Int]
_dilation Int
_groups Bool
_benchmark Bool
_deterministic = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       x9 cx9 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable x9 cx9, Castable y cy) =>
(ca
 -> cx1
 -> cx2
 -> cx3
 -> cx4
 -> cx5
 -> cx6
 -> cx7
 -> cx8
 -> cx9
 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> x8 -> x9 -> IO y
cast10 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> Int64
-> CBool
-> CBool
-> IO (ForeignPtr Tensor)
ATen.miopen_convolution_transpose_tttlllllbb) Tensor
_self Tensor
_weight Tensor
_bias [Int]
_padding [Int]
_output_padding [Int]
_stride [Int]
_dilation Int
_groups Bool
_benchmark Bool
_deterministic

miopen_depthwise_convolution ::
  -- | self
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | padding
  [Int] ->
  -- | stride
  [Int] ->
  -- | dilation
  [Int] ->
  -- | groups
  Int ->
  -- | benchmark
  Bool ->
  -- | deterministic
  Bool ->
  Tensor
miopen_depthwise_convolution :: Tensor
-> Tensor
-> Tensor
-> [Int]
-> [Int]
-> [Int]
-> Int
-> Bool
-> Bool
-> Tensor
miopen_depthwise_convolution Tensor
_self Tensor
_weight Tensor
_bias [Int]
_padding [Int]
_stride [Int]
_dilation Int
_groups Bool
_benchmark Bool
_deterministic = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable y cy) =>
(ca
 -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> cx8 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> x8 -> IO y
cast9 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> Int64
-> CBool
-> CBool
-> IO (ForeignPtr Tensor)
ATen.miopen_depthwise_convolution_tttllllbb) Tensor
_self Tensor
_weight Tensor
_bias [Int]
_padding [Int]
_stride [Int]
_dilation Int
_groups Bool
_benchmark Bool
_deterministic

miopen_rnn ::
  -- | input
  Tensor ->
  -- | weight
  [Tensor] ->
  -- | weight_stride0
  Int ->
  -- | hx
  Tensor ->
  -- | cx
  Tensor ->
  -- | mode
  Int ->
  -- | hidden_size
  Int ->
  -- | num_layers
  Int ->
  -- | batch_first
  Bool ->
  -- | dropout
  Double ->
  -- | train
  Bool ->
  -- | bidirectional
  Bool ->
  -- | batch_sizes
  [Int] ->
  -- | dropout_state
  Tensor ->
  (Tensor, Tensor, Tensor, Tensor, Tensor)
miopen_rnn :: Tensor
-> [Tensor]
-> Int
-> Tensor
-> Tensor
-> Int
-> Int
-> Int
-> Bool
-> Double
-> Bool
-> Bool
-> [Int]
-> Tensor
-> (Tensor, Tensor, Tensor, Tensor, Tensor)
miopen_rnn Tensor
_input [Tensor]
_weight Int
_weight_stride0 Tensor
_hx Tensor
_cx Int
_mode Int
_hidden_size Int
_num_layers Bool
_batch_first Double
_dropout Bool
_train Bool
_bidirectional [Int]
_batch_sizes Tensor
_dropout_state = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       x9 cx9 x10 cx10 x11 cx11 x12 cx12 x13 cx13 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable x9 cx9, Castable x10 cx10,
 Castable x11 cx11, Castable x12 cx12, Castable x13 cx13,
 Castable y cy) =>
(ca
 -> cx1
 -> cx2
 -> cx3
 -> cx4
 -> cx5
 -> cx6
 -> cx7
 -> cx8
 -> cx9
 -> cx10
 -> cx11
 -> cx12
 -> cx13
 -> IO cy)
-> a
-> x1
-> x2
-> x3
-> x4
-> x5
-> x6
-> x7
-> x8
-> x9
-> x10
-> x11
-> x12
-> x13
-> IO y
cast14 ForeignPtr Tensor
-> ForeignPtr TensorList
-> Int64
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> Int64
-> Int64
-> Int64
-> CBool
-> CDouble
-> CBool
-> CBool
-> ForeignPtr IntArray
-> ForeignPtr Tensor
-> IO
     (ForeignPtr (StdTuple '(Tensor, Tensor, Tensor, Tensor, Tensor)))
ATen.miopen_rnn_tllttlllbdbblt) Tensor
_input [Tensor]
_weight Int
_weight_stride0 Tensor
_hx Tensor
_cx Int
_mode Int
_hidden_size Int
_num_layers Bool
_batch_first Double
_dropout Bool
_train Bool
_bidirectional [Int]
_batch_sizes Tensor
_dropout_state

mm ::
  -- | self
  Tensor ->
  -- | mat2
  Tensor ->
  Tensor
mm :: Tensor -> Tensor -> Tensor
mm Tensor
_self Tensor
_mat2 = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.mm_tt) Tensor
_self Tensor
_mat2

mode ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  (Tensor, Tensor)
mode :: Tensor -> Int -> Bool -> (Tensor, Tensor)
mode Tensor
_self Int
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> Int64 -> CBool -> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.mode_tlb) Tensor
_self Int
_dim Bool
_keepdim

modeWithDimname ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  -- | keepdim
  Bool ->
  (Tensor, Tensor)
modeWithDimname :: Tensor -> Dimname -> Bool -> (Tensor, Tensor)
modeWithDimname Tensor
_self Dimname
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Dimname
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.mode_tnb) Tensor
_self Dimname
_dim Bool
_keepdim

mul ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
mul :: Tensor -> Tensor -> Tensor
mul Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.mul_tt) Tensor
_self Tensor
_other

mulScalar ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
mulScalar :: Tensor -> Float -> Tensor
mulScalar Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.mul_ts) Tensor
_self Float
_other

multiply_tt ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
multiply_tt :: Tensor -> Tensor -> Tensor
multiply_tt Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.multiply_tt) Tensor
_self Tensor
_other

multiply_ts ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
multiply_ts :: Tensor -> Float -> Tensor
multiply_ts Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.multiply_ts) Tensor
_self Float
_other

mv ::
  -- | self
  Tensor ->
  -- | vec
  Tensor ->
  Tensor
mv :: Tensor -> Tensor -> Tensor
mv Tensor
_self Tensor
_vec = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.mv_tt) Tensor
_self Tensor
_vec

mvlgamma ::
  -- | self
  Tensor ->
  -- | p
  Int ->
  Tensor
mvlgamma :: Tensor -> Int -> Tensor
mvlgamma Tensor
_self Int
_p = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.mvlgamma_tl) Tensor
_self Int
_p

narrow_copy ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | start
  Int ->
  -- | length
  Int ->
  Tensor
narrow_copy :: Tensor -> Int -> Int -> Int -> Tensor
narrow_copy Tensor
_self Int
_dim Int
_start Int
_length = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64 -> Int64 -> Int64 -> IO (ForeignPtr Tensor)
ATen.narrow_copy_tlll) Tensor
_self Int
_dim Int
_start Int
_length

narrow_tlll ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | start
  Int ->
  -- | length
  Int ->
  Tensor
narrow_tlll :: Tensor -> Int -> Int -> Int -> Tensor
narrow_tlll Tensor
_self Int
_dim Int
_start Int
_length = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64 -> Int64 -> Int64 -> IO (ForeignPtr Tensor)
ATen.narrow_tlll) Tensor
_self Int
_dim Int
_start Int
_length

narrow_tltl ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | start
  Tensor ->
  -- | length
  Int ->
  Tensor
narrow_tltl :: Tensor -> Int -> Tensor -> Int -> Tensor
narrow_tltl Tensor
_self Int
_dim Tensor
_start Int
_length = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64 -> ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.narrow_tltl) Tensor
_self Int
_dim Tensor
_start Int
_length

native_batch_norm ::
  -- | input
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | running_mean
  Tensor ->
  -- | running_var
  Tensor ->
  -- | training
  Bool ->
  -- | momentum
  Double ->
  -- | eps
  Double ->
  (Tensor, Tensor, Tensor)
native_batch_norm :: Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Bool
-> Double
-> Double
-> (Tensor, Tensor, Tensor)
native_batch_norm Tensor
_input Tensor
_weight Tensor
_bias Tensor
_running_mean Tensor
_running_var Bool
_training Double
_momentum Double
_eps = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> IO y
cast8 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> CBool
-> CDouble
-> CDouble
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor, Tensor)))
ATen.native_batch_norm_tttttbdd) Tensor
_input Tensor
_weight Tensor
_bias Tensor
_running_mean Tensor
_running_var Bool
_training Double
_momentum Double
_eps

batch_norm_stats ::
  -- | input
  Tensor ->
  -- | eps
  Double ->
  (Tensor, Tensor)
batch_norm_stats :: Tensor -> Double -> (Tensor, Tensor)
batch_norm_stats Tensor
_input Double
_eps = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor
-> CDouble -> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.batch_norm_stats_td) Tensor
_input Double
_eps

batch_norm_elemt ::
  -- | input
  Tensor ->
  -- | weight
  Tensor ->
  -- | bias
  Tensor ->
  -- | mean
  Tensor ->
  -- | invstd
  Tensor ->
  -- | eps
  Double ->
  Tensor
batch_norm_elemt :: Tensor -> Tensor -> Tensor -> Tensor -> Tensor -> Double -> Tensor
batch_norm_elemt Tensor
_input Tensor
_weight Tensor
_bias Tensor
_mean Tensor
_invstd Double
_eps = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> CDouble
-> IO (ForeignPtr Tensor)
ATen.batch_norm_elemt_tttttd) Tensor
_input Tensor
_weight Tensor
_bias Tensor
_mean Tensor
_invstd Double
_eps

batch_norm_gather_stats ::
  -- | input
  Tensor ->
  -- | mean
  Tensor ->
  -- | invstd
  Tensor ->
  -- | running_mean
  Tensor ->
  -- | running_var
  Tensor ->
  -- | momentum
  Double ->
  -- | eps
  Double ->
  -- | count
  Int ->
  (Tensor, Tensor)
batch_norm_gather_stats :: Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Double
-> Double
-> Int
-> (Tensor, Tensor)
batch_norm_gather_stats Tensor
_input Tensor
_mean Tensor
_invstd Tensor
_running_mean Tensor
_running_var Double
_momentum Double
_eps Int
_count = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> IO y
cast8 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> CDouble
-> CDouble
-> Int64
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.batch_norm_gather_stats_tttttddl) Tensor
_input Tensor
_mean Tensor
_invstd Tensor
_running_mean Tensor
_running_var Double
_momentum Double
_eps Int
_count

batch_norm_gather_stats_with_counts ::
  -- | input
  Tensor ->
  -- | mean
  Tensor ->
  -- | invstd
  Tensor ->
  -- | running_mean
  Tensor ->
  -- | running_var
  Tensor ->
  -- | momentum
  Double ->
  -- | eps
  Double ->
  -- | counts
  Tensor ->
  (Tensor, Tensor)
batch_norm_gather_stats_with_counts :: Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Double
-> Double
-> Tensor
-> (Tensor, Tensor)
batch_norm_gather_stats_with_counts Tensor
_input Tensor
_mean Tensor
_invstd Tensor
_running_mean Tensor
_running_var Double
_momentum Double
_eps Tensor
_counts = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> IO y
cast8 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> CDouble
-> CDouble
-> ForeignPtr Tensor
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.batch_norm_gather_stats_with_counts_tttttddt) Tensor
_input Tensor
_mean Tensor
_invstd Tensor
_running_mean Tensor
_running_var Double
_momentum Double
_eps Tensor
_counts

batch_norm_backward_reduce ::
  -- | grad_out
  Tensor ->
  -- | input
  Tensor ->
  -- | mean
  Tensor ->
  -- | invstd
  Tensor ->
  -- | weight
  Tensor ->
  -- | input_g
  Bool ->
  -- | weight_g
  Bool ->
  -- | bias_g
  Bool ->
  (Tensor, Tensor, Tensor, Tensor)
batch_norm_backward_reduce :: Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Bool
-> Bool
-> Bool
-> (Tensor, Tensor, Tensor, Tensor)
batch_norm_backward_reduce Tensor
_grad_out Tensor
_input Tensor
_mean Tensor
_invstd Tensor
_weight Bool
_input_g Bool
_weight_g Bool
_bias_g = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> IO y
cast8 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> CBool
-> CBool
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor, Tensor, Tensor)))
ATen.batch_norm_backward_reduce_tttttbbb) Tensor
_grad_out Tensor
_input Tensor
_mean Tensor
_invstd Tensor
_weight Bool
_input_g Bool
_weight_g Bool
_bias_g

batch_norm_backward_elemt ::
  -- | grad_out
  Tensor ->
  -- | input
  Tensor ->
  -- | mean
  Tensor ->
  -- | invstd
  Tensor ->
  -- | weight
  Tensor ->
  -- | mean_dy
  Tensor ->
  -- | mean_dy_xmu
  Tensor ->
  -- | count
  Tensor ->
  Tensor
batch_norm_backward_elemt :: Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
batch_norm_backward_elemt Tensor
_grad_out Tensor
_input Tensor
_mean Tensor
_invstd Tensor
_weight Tensor
_mean_dy Tensor
_mean_dy_xmu Tensor
_count = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> IO y
cast8 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> IO (ForeignPtr Tensor)
ATen.batch_norm_backward_elemt_tttttttt) Tensor
_grad_out Tensor
_input Tensor
_mean Tensor
_invstd Tensor
_weight Tensor
_mean_dy Tensor
_mean_dy_xmu Tensor
_count

batch_norm_update_stats ::
  -- | input
  Tensor ->
  -- | running_mean
  Tensor ->
  -- | running_var
  Tensor ->
  -- | momentum
  Double ->
  (Tensor, Tensor)
batch_norm_update_stats :: Tensor -> Tensor -> Tensor -> Double -> (Tensor, Tensor)
batch_norm_update_stats Tensor
_input Tensor
_running_mean Tensor
_running_var Double
_momentum = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> CDouble
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.batch_norm_update_stats_tttd) Tensor
_input Tensor
_running_mean Tensor
_running_var Double
_momentum

is_vulkan_available ::
  Bool
is_vulkan_available :: Bool
is_vulkan_available = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca. Castable a ca => IO ca -> IO a
cast0 IO CBool
ATen.is_vulkan_available)

pairwise_distance ::
  -- | x1
  Tensor ->
  -- | x2
  Tensor ->
  -- | p
  Double ->
  -- | eps
  Double ->
  -- | keepdim
  Bool ->
  Tensor
pairwise_distance :: Tensor -> Tensor -> Double -> Double -> Bool -> Tensor
pairwise_distance Tensor
_x1 Tensor
_x2 Double
_p Double
_eps Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> CDouble
-> CDouble
-> CBool
-> IO (ForeignPtr Tensor)
ATen.pairwise_distance_ttddb) Tensor
_x1 Tensor
_x2 Double
_p Double
_eps Bool
_keepdim

cdist ::
  -- | x1
  Tensor ->
  -- | x2
  Tensor ->
  -- | p
  Double ->
  -- | compute_mode
  Int ->
  Tensor
cdist :: Tensor -> Tensor -> Double -> Int -> Tensor
cdist Tensor
_x1 Tensor
_x2 Double
_p Int
_compute_mode = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Tensor -> CDouble -> Int64 -> IO (ForeignPtr Tensor)
ATen.cdist_ttdl) Tensor
_x1 Tensor
_x2 Double
_p Int
_compute_mode

pdist ::
  -- | self
  Tensor ->
  -- | p
  Double ->
  Tensor
pdist :: Tensor -> Double -> Tensor
pdist Tensor
_self Double
_p = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> CDouble -> IO (ForeignPtr Tensor)
ATen.pdist_td) Tensor
_self Double
_p

cosine_similarity ::
  -- | x1
  Tensor ->
  -- | x2
  Tensor ->
  -- | dim
  Int ->
  -- | eps
  Double ->
  Tensor
cosine_similarity :: Tensor -> Tensor -> Int -> Double -> Tensor
cosine_similarity Tensor
_x1 Tensor
_x2 Int
_dim Double
_eps = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Tensor -> Int64 -> CDouble -> IO (ForeignPtr Tensor)
ATen.cosine_similarity_ttld) Tensor
_x1 Tensor
_x2 Int
_dim Double
_eps

permute ::
  -- | self
  Tensor ->
  -- | dims
  [Int] ->
  Tensor
permute :: Tensor -> [Int] -> Tensor
permute Tensor
_self [Int]
_dims = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr IntArray -> IO (ForeignPtr Tensor)
ATen.permute_tl) Tensor
_self [Int]
_dims

movedim_tll ::
  -- | self
  Tensor ->
  -- | source
  [Int] ->
  -- | destination
  [Int] ->
  Tensor
movedim_tll :: Tensor -> [Int] -> [Int] -> Tensor
movedim_tll Tensor
_self [Int]
_source [Int]
_destination = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> IO (ForeignPtr Tensor)
ATen.movedim_tll) Tensor
_self [Int]
_source [Int]
_destination

-- movedim_tll
--   :: Tensor -- ^ self
--   -> Int -- ^ source
--   -> Int -- ^ destination
--   -> Tensor
-- movedim_tll _self _source _destination = unsafePerformIO $ (cast3 ATen.movedim_tll) _self _source _destination

moveaxis_tll ::
  -- | self
  Tensor ->
  -- | source
  [Int] ->
  -- | destination
  [Int] ->
  Tensor
moveaxis_tll :: Tensor -> [Int] -> [Int] -> Tensor
moveaxis_tll Tensor
_self [Int]
_source [Int]
_destination = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> IO (ForeignPtr Tensor)
ATen.moveaxis_tll) Tensor
_self [Int]
_source [Int]
_destination

-- moveaxis_tll
--   :: Tensor -- ^ self
--   -> Int -- ^ source
--   -> Int -- ^ destination
--   -> Tensor
-- moveaxis_tll _self _source _destination = unsafePerformIO $ (cast3 ATen.moveaxis_tll) _self _source _destination

adjoint ::
  -- | self
  Tensor ->
  Tensor
adjoint :: Tensor -> Tensor
adjoint Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.adjoint_t) Tensor
_self

pixel_shuffle ::
  -- | self
  Tensor ->
  -- | upscale_factor
  Int ->
  Tensor
pixel_shuffle :: Tensor -> Int -> Tensor
pixel_shuffle Tensor
_self Int
_upscale_factor = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.pixel_shuffle_tl) Tensor
_self Int
_upscale_factor

pixel_unshuffle ::
  -- | self
  Tensor ->
  -- | downscale_factor
  Int ->
  Tensor
pixel_unshuffle :: Tensor -> Int -> Tensor
pixel_unshuffle Tensor
_self Int
_downscale_factor = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.pixel_unshuffle_tl) Tensor
_self Int
_downscale_factor

channel_shuffle ::
  -- | self
  Tensor ->
  -- | groups
  Int ->
  Tensor
channel_shuffle :: Tensor -> Int -> Tensor
channel_shuffle Tensor
_self Int
_groups = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.channel_shuffle_tl) Tensor
_self Int
_groups

native_channel_shuffle ::
  -- | self
  Tensor ->
  -- | groups
  Int ->
  Tensor
native_channel_shuffle :: Tensor -> Int -> Tensor
native_channel_shuffle Tensor
_self Int
_groups = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.native_channel_shuffle_tl) Tensor
_self Int
_groups

pinverse ::
  -- | self
  Tensor ->
  -- | rcond
  Double ->
  Tensor
pinverse :: Tensor -> Double -> Tensor
pinverse Tensor
_self Double
_rcond = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> CDouble -> IO (ForeignPtr Tensor)
ATen.pinverse_td) Tensor
_self Double
_rcond

poisson_nll_loss ::
  -- | input
  Tensor ->
  -- | target
  Tensor ->
  -- | log_input
  Bool ->
  -- | full
  Bool ->
  -- | eps
  Double ->
  -- | reduction
  Int ->
  Tensor
poisson_nll_loss :: Tensor -> Tensor -> Bool -> Bool -> Double -> Int -> Tensor
poisson_nll_loss Tensor
_input Tensor
_target Bool
_log_input Bool
_full Double
_eps Int
_reduction = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr Tensor
-> CBool
-> CBool
-> CDouble
-> Int64
-> IO (ForeignPtr Tensor)
ATen.poisson_nll_loss_ttbbdl) Tensor
_input Tensor
_target Bool
_log_input Bool
_full Double
_eps Int
_reduction

rad2deg ::
  -- | self
  Tensor ->
  Tensor
rad2deg :: Tensor -> Tensor
rad2deg Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.rad2deg_t) Tensor
_self

deg2rad ::
  -- | self
  Tensor ->
  Tensor
deg2rad :: Tensor -> Tensor
deg2rad Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.deg2rad_t) Tensor
_self

ravel ::
  -- | self
  Tensor ->
  Tensor
ravel :: Tensor -> Tensor
ravel Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.ravel_t) Tensor
_self

reciprocal ::
  -- | self
  Tensor ->
  Tensor
reciprocal :: Tensor -> Tensor
reciprocal Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.reciprocal_t) Tensor
_self

neg ::
  -- | self
  Tensor ->
  Tensor
neg :: Tensor -> Tensor
neg Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.neg_t) Tensor
_self

negative ::
  -- | self
  Tensor ->
  Tensor
negative :: Tensor -> Tensor
negative Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.negative_t) Tensor
_self

repeat_interleave_tl ::
  -- | repeats
  Tensor ->
  -- | output_size
  Int ->
  Tensor
repeat_interleave_tl :: Tensor -> Int -> Tensor
repeat_interleave_tl Tensor
_repeats Int
_output_size = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.repeat_interleave_tl) Tensor
_repeats Int
_output_size

repeat_interleave_ttll ::
  -- | self
  Tensor ->
  -- | repeats
  Tensor ->
  -- | dim
  Int ->
  -- | output_size
  Int ->
  Tensor
repeat_interleave_ttll :: Tensor -> Tensor -> Int -> Int -> Tensor
repeat_interleave_ttll Tensor
_self Tensor
_repeats Int
_dim Int
_output_size = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Tensor -> Int64 -> Int64 -> IO (ForeignPtr Tensor)
ATen.repeat_interleave_ttll) Tensor
_self Tensor
_repeats Int
_dim Int
_output_size

repeat_interleave_tlll ::
  -- | self
  Tensor ->
  -- | repeats
  Int ->
  -- | dim
  Int ->
  -- | output_size
  Int ->
  Tensor
repeat_interleave_tlll :: Tensor -> Int -> Int -> Int -> Tensor
repeat_interleave_tlll Tensor
_self Int
_repeats Int
_dim Int
_output_size = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64 -> Int64 -> Int64 -> IO (ForeignPtr Tensor)
ATen.repeat_interleave_tlll) Tensor
_self Int
_repeats Int
_dim Int
_output_size

reshape ::
  -- | self
  Tensor ->
  -- | shape
  [Int] ->
  Tensor
reshape :: Tensor -> [Int] -> Tensor
reshape Tensor
_self [Int]
_shape = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr IntArray -> IO (ForeignPtr Tensor)
ATen.reshape_tl) Tensor
_self [Int]
_shape

round_t ::
  -- | self
  Tensor ->
  Tensor
round_t :: Tensor -> Tensor
round_t Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.round_t) Tensor
_self

round_tl ::
  -- | self
  Tensor ->
  -- | decimals
  Int ->
  Tensor
round_tl :: Tensor -> Int -> Tensor
round_tl Tensor
_self Int
_decimals = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.round_tl) Tensor
_self Int
_decimals

relu ::
  -- | self
  Tensor ->
  Tensor
relu :: Tensor -> Tensor
relu Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.relu_t) Tensor
_self

relu6 ::
  -- | self
  Tensor ->
  Tensor
relu6 :: Tensor -> Tensor
relu6 Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.relu6_t) Tensor
_self

prelu ::
  -- | self
  Tensor ->
  -- | weight
  Tensor ->
  Tensor
prelu :: Tensor -> Tensor -> Tensor
prelu Tensor
_self Tensor
_weight = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.prelu_tt) Tensor
_self Tensor
_weight

gelu ::
  -- | self
  Tensor ->
  Tensor
gelu :: Tensor -> Tensor
gelu Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.gelu_t) Tensor
_self

hardshrink ::
  -- | self
  Tensor ->
  -- | lambd
  Float ->
  Tensor
hardshrink :: Tensor -> Float -> Tensor
hardshrink Tensor
_self Float
_lambd = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.hardshrink_ts) Tensor
_self Float
_lambd

rsqrt ::
  -- | self
  Tensor ->
  Tensor
rsqrt :: Tensor -> Tensor
rsqrt Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.rsqrt_t) Tensor
_self

selectWithDimname ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  -- | index
  Int ->
  Tensor
selectWithDimname :: Tensor -> Dimname -> Int -> Tensor
selectWithDimname Tensor
_self Dimname
_dim Int
_index = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Dimname -> Int64 -> IO (ForeignPtr Tensor)
ATen.select_tnl) Tensor
_self Dimname
_dim Int
_index

select ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | index
  Int ->
  Tensor
select :: Tensor -> Int -> Int -> Tensor
select Tensor
_self Int
_dim Int
_index = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> Int64 -> Int64 -> IO (ForeignPtr Tensor)
ATen.select_tll) Tensor
_self Int
_dim Int
_index

selu ::
  -- | self
  Tensor ->
  Tensor
selu :: Tensor -> Tensor
selu Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.selu_t) Tensor
_self

celu ::
  -- | self
  Tensor ->
  -- | alpha
  Float ->
  Tensor
celu :: Tensor -> Float -> Tensor
celu Tensor
_self Float
_alpha = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.celu_ts) Tensor
_self Float
_alpha

silu ::
  -- | self
  Tensor ->
  Tensor
silu :: Tensor -> Tensor
silu Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.silu_t) Tensor
_self

mish ::
  -- | self
  Tensor ->
  Tensor
mish :: Tensor -> Tensor
mish Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.mish_t) Tensor
_self

sigmoid ::
  -- | self
  Tensor ->
  Tensor
sigmoid :: Tensor -> Tensor
sigmoid Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.sigmoid_t) Tensor
_self

logit ::
  -- | self
  Tensor ->
  -- | eps
  Double ->
  Tensor
logit :: Tensor -> Double -> Tensor
logit Tensor
_self Double
_eps = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> CDouble -> IO (ForeignPtr Tensor)
ATen.logit_td) Tensor
_self Double
_eps

sin ::
  -- | self
  Tensor ->
  Tensor
sin :: Tensor -> Tensor
sin Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.sin_t) Tensor
_self

sinc ::
  -- | self
  Tensor ->
  Tensor
sinc :: Tensor -> Tensor
sinc Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.sinc_t) Tensor
_self

sinh ::
  -- | self
  Tensor ->
  Tensor
sinh :: Tensor -> Tensor
sinh Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.sinh_t) Tensor
_self

size ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  Int
size :: Tensor -> Int -> Int
size Tensor
_self Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO Int64
ATen.size_tl) Tensor
_self Int
_dim

sizeWithDimname ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  Int
sizeWithDimname :: Tensor -> Dimname -> Int
sizeWithDimname Tensor
_self Dimname
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Dimname -> IO Int64
ATen.size_tn) Tensor
_self Dimname
_dim

slice ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | start
  Int ->
  -- | end
  Int ->
  -- | step
  Int ->
  Tensor
slice :: Tensor -> Int -> Int -> Int -> Int -> Tensor
slice Tensor
_self Int
_dim Int
_start Int
_end Int
_step = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> Int64 -> Int64 -> Int64 -> Int64 -> IO (ForeignPtr Tensor)
ATen.slice_tllll) Tensor
_self Int
_dim Int
_start Int
_end Int
_step

slice_scatter ::
  -- | self
  Tensor ->
  -- | src
  Tensor ->
  -- | dim
  Int ->
  -- | start
  Int ->
  -- | end
  Int ->
  -- | step
  Int ->
  Tensor
slice_scatter :: Tensor -> Tensor -> Int -> Int -> Int -> Int -> Tensor
slice_scatter Tensor
_self Tensor
_src Int
_dim Int
_start Int
_end Int
_step = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr Tensor
-> Int64
-> Int64
-> Int64
-> Int64
-> IO (ForeignPtr Tensor)
ATen.slice_scatter_ttllll) Tensor
_self Tensor
_src Int
_dim Int
_start Int
_end Int
_step

select_scatter ::
  -- | self
  Tensor ->
  -- | src
  Tensor ->
  -- | dim
  Int ->
  -- | index
  Int ->
  Tensor
select_scatter :: Tensor -> Tensor -> Int -> Int -> Tensor
select_scatter Tensor
_self Tensor
_src Int
_dim Int
_index = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Tensor -> Int64 -> Int64 -> IO (ForeignPtr Tensor)
ATen.select_scatter_ttll) Tensor
_self Tensor
_src Int
_dim Int
_index

diagonal_scatter ::
  -- | self
  Tensor ->
  -- | src
  Tensor ->
  -- | offset
  Int ->
  -- | dim1
  Int ->
  -- | dim2
  Int ->
  Tensor
diagonal_scatter :: Tensor -> Tensor -> Int -> Int -> Int -> Tensor
diagonal_scatter Tensor
_self Tensor
_src Int
_offset Int
_dim1 Int
_dim2 = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> Int64
-> Int64
-> Int64
-> IO (ForeignPtr Tensor)
ATen.diagonal_scatter_ttlll) Tensor
_self Tensor
_src Int
_offset Int
_dim1 Int
_dim2

slogdet ::
  -- | self
  Tensor ->
  (Tensor, Tensor)
slogdet :: Tensor -> (Tensor, Tensor)
slogdet Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.slogdet_t) Tensor
_self

smm ::
  -- | self
  Tensor ->
  -- | mat2
  Tensor ->
  Tensor
smm :: Tensor -> Tensor -> Tensor
smm Tensor
_self Tensor
_mat2 = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.smm_tt) Tensor
_self Tensor
_mat2

softmax ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | dtype
  DType ->
  Tensor
softmax :: Tensor -> Int -> DType -> Tensor
softmax Tensor
_self Int
_dim DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> Int64 -> ScalarType -> IO (ForeignPtr Tensor)
ATen.softmax_tls) Tensor
_self Int
_dim DType
_dtype

softmaxWithDimname ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  -- | dtype
  DType ->
  Tensor
softmaxWithDimname :: Tensor -> Dimname -> DType -> Tensor
softmaxWithDimname Tensor
_self Dimname
_dim DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Dimname -> ScalarType -> IO (ForeignPtr Tensor)
ATen.softmax_tns) Tensor
_self Dimname
_dim DType
_dtype

unsafe_split ::
  -- | self
  Tensor ->
  -- | split_size
  Int ->
  -- | dim
  Int ->
  [Tensor]
unsafe_split :: Tensor -> Int -> Int -> [Tensor]
unsafe_split Tensor
_self Int
_split_size Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> Int64 -> Int64 -> IO (ForeignPtr TensorList)
ATen.unsafe_split_tll) Tensor
_self Int
_split_size Int
_dim

split ::
  -- | self
  Tensor ->
  -- | split_size
  Int ->
  -- | dim
  Int ->
  [Tensor]
split :: Tensor -> Int -> Int -> [Tensor]
split Tensor
_self Int
_split_size Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> Int64 -> Int64 -> IO (ForeignPtr TensorList)
ATen.split_tll) Tensor
_self Int
_split_size Int
_dim

unsafe_split_with_sizes ::
  -- | self
  Tensor ->
  -- | split_sizes
  [Int] ->
  -- | dim
  Int ->
  [Tensor]
unsafe_split_with_sizes :: Tensor -> [Int] -> Int -> [Tensor]
unsafe_split_with_sizes Tensor
_self [Int]
_split_sizes Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr IntArray -> Int64 -> IO (ForeignPtr TensorList)
ATen.unsafe_split_with_sizes_tll) Tensor
_self [Int]
_split_sizes Int
_dim

split_with_sizes ::
  -- | self
  Tensor ->
  -- | split_sizes
  [Int] ->
  -- | dim
  Int ->
  [Tensor]
split_with_sizes :: Tensor -> [Int] -> Int -> [Tensor]
split_with_sizes Tensor
_self [Int]
_split_sizes Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr IntArray -> Int64 -> IO (ForeignPtr TensorList)
ATen.split_with_sizes_tll) Tensor
_self [Int]
_split_sizes Int
_dim

hsplit_tl ::
  -- | self
  Tensor ->
  -- | sections
  Int ->
  [Tensor]
hsplit_tl :: Tensor -> Int -> [Tensor]
hsplit_tl Tensor
_self Int
_sections = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO (ForeignPtr TensorList)
ATen.hsplit_tl) Tensor
_self Int
_sections

-- hsplit_tl
--   :: Tensor -- ^ self
--   -> [Int] -- ^ indices
--   -> [Tensor]
-- hsplit_tl _self _indices = unsafePerformIO $ (cast2 ATen.hsplit_tl) _self _indices

vsplit_tl ::
  -- | self
  Tensor ->
  -- | sections
  Int ->
  [Tensor]
vsplit_tl :: Tensor -> Int -> [Tensor]
vsplit_tl Tensor
_self Int
_sections = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO (ForeignPtr TensorList)
ATen.vsplit_tl) Tensor
_self Int
_sections

-- vsplit_tl
--   :: Tensor -- ^ self
--   -> [Int] -- ^ indices
--   -> [Tensor]
-- vsplit_tl _self _indices = unsafePerformIO $ (cast2 ATen.vsplit_tl) _self _indices

dsplit_tl ::
  -- | self
  Tensor ->
  -- | sections
  Int ->
  [Tensor]
dsplit_tl :: Tensor -> Int -> [Tensor]
dsplit_tl Tensor
_self Int
_sections = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO (ForeignPtr TensorList)
ATen.dsplit_tl) Tensor
_self Int
_sections

-- dsplit_tl
--   :: Tensor -- ^ self
--   -> [Int] -- ^ indices
--   -> [Tensor]
-- dsplit_tl _self _indices = unsafePerformIO $ (cast2 ATen.dsplit_tl) _self _indices

squeezeAll ::
  -- | self
  Tensor ->
  Tensor
squeezeAll :: Tensor -> Tensor
squeezeAll Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.squeeze_t) Tensor
_self

squeezeDim ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  Tensor
squeezeDim :: Tensor -> Int -> Tensor
squeezeDim Tensor
_self Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.squeeze_tl) Tensor
_self Int
_dim

squeezeWithDimname ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  Tensor
squeezeWithDimname :: Tensor -> Dimname -> Tensor
squeezeWithDimname Tensor
_self Dimname
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Dimname -> IO (ForeignPtr Tensor)
ATen.squeeze_tn) Tensor
_self Dimname
_dim

sspaddmm ::
  -- | self
  Tensor ->
  -- | mat1
  Tensor ->
  -- | mat2
  Tensor ->
  -- | beta
  Float ->
  -- | alpha
  Float ->
  Tensor
sspaddmm :: Tensor -> Tensor -> Tensor -> Float -> Float -> Tensor
sspaddmm Tensor
_self Tensor
_mat1 Tensor
_mat2 Float
_beta Float
_alpha = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr Scalar
-> IO (ForeignPtr Tensor)
ATen.sspaddmm_tttss) Tensor
_self Tensor
_mat1 Tensor
_mat2 Float
_beta Float
_alpha

stack ::
  -- | tensors
  [Tensor] ->
  -- | dim
  Int ->
  Tensor
stack :: [Tensor] -> Int -> Tensor
stack [Tensor]
_tensors Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr TensorList -> Int64 -> IO (ForeignPtr Tensor)
ATen.stack_ll) [Tensor]
_tensors Int
_dim

hstack ::
  -- | tensors
  [Tensor] ->
  Tensor
hstack :: [Tensor] -> Tensor
hstack [Tensor]
_tensors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr TensorList -> IO (ForeignPtr Tensor)
ATen.hstack_l) [Tensor]
_tensors

vstack ::
  -- | tensors
  [Tensor] ->
  Tensor
vstack :: [Tensor] -> Tensor
vstack [Tensor]
_tensors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr TensorList -> IO (ForeignPtr Tensor)
ATen.vstack_l) [Tensor]
_tensors

dstack ::
  -- | tensors
  [Tensor] ->
  Tensor
dstack :: [Tensor] -> Tensor
dstack [Tensor]
_tensors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr TensorList -> IO (ForeignPtr Tensor)
ATen.dstack_l) [Tensor]
_tensors

stft ::
  -- | self
  Tensor ->
  -- | n_fft
  Int ->
  -- | hop_length
  Int ->
  -- | win_length
  Int ->
  -- | window
  Tensor ->
  -- | normalized
  Bool ->
  -- | onesided
  Bool ->
  -- | return_complex
  Bool ->
  Tensor
stft :: Tensor
-> Int -> Int -> Int -> Tensor -> Bool -> Bool -> Bool -> Tensor
stft Tensor
_self Int
_n_fft Int
_hop_length Int
_win_length Tensor
_window Bool
_normalized Bool
_onesided Bool
_return_complex = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> IO y
cast8 ForeignPtr Tensor
-> Int64
-> Int64
-> Int64
-> ForeignPtr Tensor
-> CBool
-> CBool
-> CBool
-> IO (ForeignPtr Tensor)
ATen.stft_tllltbbb) Tensor
_self Int
_n_fft Int
_hop_length Int
_win_length Tensor
_window Bool
_normalized Bool
_onesided Bool
_return_complex

istft ::
  -- | self
  Tensor ->
  -- | n_fft
  Int ->
  -- | hop_length
  Int ->
  -- | win_length
  Int ->
  -- | window
  Tensor ->
  -- | center
  Bool ->
  -- | normalized
  Bool ->
  -- | onesided
  Bool ->
  -- | length
  Int ->
  -- | return_complex
  Bool ->
  Tensor
istft :: Tensor
-> Int
-> Int
-> Int
-> Tensor
-> Bool
-> Bool
-> Bool
-> Int
-> Bool
-> Tensor
istft Tensor
_self Int
_n_fft Int
_hop_length Int
_win_length Tensor
_window Bool
_center Bool
_normalized Bool
_onesided Int
_length Bool
_return_complex = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       x9 cx9 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable x9 cx9, Castable y cy) =>
(ca
 -> cx1
 -> cx2
 -> cx3
 -> cx4
 -> cx5
 -> cx6
 -> cx7
 -> cx8
 -> cx9
 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> x8 -> x9 -> IO y
cast10 ForeignPtr Tensor
-> Int64
-> Int64
-> Int64
-> ForeignPtr Tensor
-> CBool
-> CBool
-> CBool
-> Int64
-> CBool
-> IO (ForeignPtr Tensor)
ATen.istft_tllltbbblb) Tensor
_self Int
_n_fft Int
_hop_length Int
_win_length Tensor
_window Bool
_center Bool
_normalized Bool
_onesided Int
_length Bool
_return_complex

stride ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  Int
stride :: Tensor -> Int -> Int
stride Tensor
_self Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO Int64
ATen.stride_tl) Tensor
_self Int
_dim

strideWithDimname ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  Int
strideWithDimname :: Tensor -> Dimname -> Int
strideWithDimname Tensor
_self Dimname
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Dimname -> IO Int64
ATen.stride_tn) Tensor
_self Dimname
_dim

sumAll ::
  -- | self
  Tensor ->
  -- | dtype
  DType ->
  Tensor
sumAll :: Tensor -> DType -> Tensor
sumAll Tensor
_self DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ScalarType -> IO (ForeignPtr Tensor)
ATen.sum_ts) Tensor
_self DType
_dtype

sumDim ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  -- | dtype
  DType ->
  Tensor
sumDim :: Tensor -> Int -> Bool -> DType -> Tensor
sumDim Tensor
_self Int
_dim Bool
_keepdim DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> CBool
-> ScalarType
-> IO (ForeignPtr Tensor)
ATen.sum_tlbs) Tensor
_self Int
_dim Bool
_keepdim DType
_dtype

sumWithDimnames ::
  -- | self
  Tensor ->
  -- | dim
  [Dimname] ->
  -- | keepdim
  Bool ->
  -- | dtype
  DType ->
  Tensor
sumWithDimnames :: Tensor -> [Dimname] -> Bool -> DType -> Tensor
sumWithDimnames Tensor
_self [Dimname]
_dim Bool
_keepdim DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr DimnameList
-> CBool
-> ScalarType
-> IO (ForeignPtr Tensor)
ATen.sum_tNbs) Tensor
_self [Dimname]
_dim Bool
_keepdim DType
_dtype

nansum_ts ::
  -- | self
  Tensor ->
  -- | dtype
  DType ->
  Tensor
nansum_ts :: Tensor -> DType -> Tensor
nansum_ts Tensor
_self DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ScalarType -> IO (ForeignPtr Tensor)
ATen.nansum_ts) Tensor
_self DType
_dtype

nansum_tlbs ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  -- | dtype
  DType ->
  Tensor
nansum_tlbs :: Tensor -> Int -> Bool -> DType -> Tensor
nansum_tlbs Tensor
_self Int
_dim Bool
_keepdim DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> CBool
-> ScalarType
-> IO (ForeignPtr Tensor)
ATen.nansum_tlbs) Tensor
_self Int
_dim Bool
_keepdim DType
_dtype

sqrt ::
  -- | self
  Tensor ->
  Tensor
sqrt :: Tensor -> Tensor
sqrt Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.sqrt_t) Tensor
_self

square ::
  -- | self
  Tensor ->
  Tensor
square :: Tensor -> Tensor
square Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.square_t) Tensor
_self

stdAll ::
  -- | self
  Tensor ->
  -- | unbiased
  Bool ->
  Tensor
stdAll :: Tensor -> Bool -> Tensor
stdAll Tensor
_self Bool
_unbiased = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> CBool -> IO (ForeignPtr Tensor)
ATen.std_tb) Tensor
_self Bool
_unbiased

stdDim ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | unbiased
  Bool ->
  -- | keepdim
  Bool ->
  Tensor
stdDim :: Tensor -> Int -> Bool -> Bool -> Tensor
stdDim Tensor
_self Int
_dim Bool
_unbiased Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray -> CBool -> CBool -> IO (ForeignPtr Tensor)
ATen.std_tlbb) Tensor
_self Int
_dim Bool
_unbiased Bool
_keepdim

std_tllb ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | correction
  Int ->
  -- | keepdim
  Bool ->
  Tensor
std_tllb :: Tensor -> Int -> Int -> Bool -> Tensor
std_tllb Tensor
_self Int
_dim Int
_correction Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray -> Int64 -> CBool -> IO (ForeignPtr Tensor)
ATen.std_tllb) Tensor
_self Int
_dim Int
_correction Bool
_keepdim

stdMeanAll ::
  -- | self
  Tensor ->
  -- | unbiased
  Bool ->
  (Tensor, Tensor)
stdMeanAll :: Tensor -> Bool -> (Tensor, Tensor)
stdMeanAll Tensor
_self Bool
_unbiased = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor
-> CBool -> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.std_mean_tb) Tensor
_self Bool
_unbiased

stdMeanDim ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | unbiased
  Bool ->
  -- | keepdim
  Bool ->
  (Tensor, Tensor)
stdMeanDim :: Tensor -> Int -> Bool -> Bool -> (Tensor, Tensor)
stdMeanDim Tensor
_self Int
_dim Bool
_unbiased Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> CBool
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.std_mean_tlbb) Tensor
_self Int
_dim Bool
_unbiased Bool
_keepdim

std_mean_tllb ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | correction
  Int ->
  -- | keepdim
  Bool ->
  (Tensor, Tensor)
std_mean_tllb :: Tensor -> Int -> Int -> Bool -> (Tensor, Tensor)
std_mean_tllb Tensor
_self Int
_dim Int
_correction Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> Int64
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.std_mean_tllb) Tensor
_self Int
_dim Int
_correction Bool
_keepdim

stdMeanWithDimnames ::
  -- | self
  Tensor ->
  -- | dim
  [Dimname] ->
  -- | unbiased
  Bool ->
  -- | keepdim
  Bool ->
  (Tensor, Tensor)
stdMeanWithDimnames :: Tensor -> [Dimname] -> Bool -> Bool -> (Tensor, Tensor)
stdMeanWithDimnames Tensor
_self [Dimname]
_dim Bool
_unbiased Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr DimnameList
-> CBool
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.std_mean_tNbb) Tensor
_self [Dimname]
_dim Bool
_unbiased Bool
_keepdim

std_mean_tNlb ::
  -- | self
  Tensor ->
  -- | dim
  [Dimname] ->
  -- | correction
  Int ->
  -- | keepdim
  Bool ->
  (Tensor, Tensor)
std_mean_tNlb :: Tensor -> [Dimname] -> Int -> Bool -> (Tensor, Tensor)
std_mean_tNlb Tensor
_self [Dimname]
_dim Int
_correction Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr DimnameList
-> Int64
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.std_mean_tNlb) Tensor
_self [Dimname]
_dim Int
_correction Bool
_keepdim

stdWithDimnames ::
  -- | self
  Tensor ->
  -- | dim
  [Dimname] ->
  -- | unbiased
  Bool ->
  -- | keepdim
  Bool ->
  Tensor
stdWithDimnames :: Tensor -> [Dimname] -> Bool -> Bool -> Tensor
stdWithDimnames Tensor
_self [Dimname]
_dim Bool
_unbiased Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr DimnameList
-> CBool
-> CBool
-> IO (ForeignPtr Tensor)
ATen.std_tNbb) Tensor
_self [Dimname]
_dim Bool
_unbiased Bool
_keepdim

std_tNlb ::
  -- | self
  Tensor ->
  -- | dim
  [Dimname] ->
  -- | correction
  Int ->
  -- | keepdim
  Bool ->
  Tensor
std_tNlb :: Tensor -> [Dimname] -> Int -> Bool -> Tensor
std_tNlb Tensor
_self [Dimname]
_dim Int
_correction Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr DimnameList
-> Int64
-> CBool
-> IO (ForeignPtr Tensor)
ATen.std_tNlb) Tensor
_self [Dimname]
_dim Int
_correction Bool
_keepdim

prodAll ::
  -- | self
  Tensor ->
  -- | dtype
  DType ->
  Tensor
prodAll :: Tensor -> DType -> Tensor
prodAll Tensor
_self DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ScalarType -> IO (ForeignPtr Tensor)
ATen.prod_ts) Tensor
_self DType
_dtype

prodDim ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  -- | dtype
  DType ->
  Tensor
prodDim :: Tensor -> Int -> Bool -> DType -> Tensor
prodDim Tensor
_self Int
_dim Bool
_keepdim DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64 -> CBool -> ScalarType -> IO (ForeignPtr Tensor)
ATen.prod_tlbs) Tensor
_self Int
_dim Bool
_keepdim DType
_dtype

prodWithDimnames ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  -- | keepdim
  Bool ->
  -- | dtype
  DType ->
  Tensor
prodWithDimnames :: Tensor -> Dimname -> Bool -> DType -> Tensor
prodWithDimnames Tensor
_self Dimname
_dim Bool
_keepdim DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Dimname
-> CBool
-> ScalarType
-> IO (ForeignPtr Tensor)
ATen.prod_tnbs) Tensor
_self Dimname
_dim Bool
_keepdim DType
_dtype

t ::
  -- | self
  Tensor ->
  Tensor
t :: Tensor -> Tensor
t Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.t_t) Tensor
_self

tan ::
  -- | self
  Tensor ->
  Tensor
tan :: Tensor -> Tensor
tan Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.tan_t) Tensor
_self

tanh ::
  -- | self
  Tensor ->
  Tensor
tanh :: Tensor -> Tensor
tanh Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.tanh_t) Tensor
_self

tensordot ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  -- | dims_self
  [Int] ->
  -- | dims_other
  [Int] ->
  Tensor
tensordot :: Tensor -> Tensor -> [Int] -> [Int] -> Tensor
tensordot Tensor
_self Tensor
_other [Int]
_dims_self [Int]
_dims_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> IO (ForeignPtr Tensor)
ATen.tensordot_ttll) Tensor
_self Tensor
_other [Int]
_dims_self [Int]
_dims_other

threshold ::
  -- | self
  Tensor ->
  -- | threshold
  Float ->
  -- | value
  Float ->
  Tensor
threshold :: Tensor -> Float -> Float -> Tensor
threshold Tensor
_self Float
_threshold Float
_value = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Scalar -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.threshold_tss) Tensor
_self Float
_threshold Float
_value

tile ::
  -- | self
  Tensor ->
  -- | dims
  [Int] ->
  Tensor
tile :: Tensor -> [Int] -> Tensor
tile Tensor
_self [Int]
_dims = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr IntArray -> IO (ForeignPtr Tensor)
ATen.tile_tl) Tensor
_self [Int]
_dims

transpose ::
  -- | self
  Tensor ->
  -- | dim0
  Int ->
  -- | dim1
  Int ->
  Tensor
transpose :: Tensor -> Int -> Int -> Tensor
transpose Tensor
_self Int
_dim0 Int
_dim1 = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> Int64 -> Int64 -> IO (ForeignPtr Tensor)
ATen.transpose_tll) Tensor
_self Int
_dim0 Int
_dim1

transposeWithDimname ::
  -- | self
  Tensor ->
  -- | dim0
  Dimname ->
  -- | dim1
  Dimname ->
  Tensor
transposeWithDimname :: Tensor -> Dimname -> Dimname -> Tensor
transposeWithDimname Tensor
_self Dimname
_dim0 Dimname
_dim1 = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Dimname
-> ForeignPtr Dimname
-> IO (ForeignPtr Tensor)
ATen.transpose_tnn) Tensor
_self Dimname
_dim0 Dimname
_dim1

one_hot ::
  -- | self
  Tensor ->
  -- | num_classes
  Int ->
  Tensor
one_hot :: Tensor -> Int -> Tensor
one_hot Tensor
_self Int
_num_classes = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.one_hot_tl) Tensor
_self Int
_num_classes

flip ::
  -- | self
  Tensor ->
  -- | dims
  [Int] ->
  Tensor
flip :: Tensor -> [Int] -> Tensor
flip Tensor
_self [Int]
_dims = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr IntArray -> IO (ForeignPtr Tensor)
ATen.flip_tl) Tensor
_self [Int]
_dims

fliplr ::
  -- | self
  Tensor ->
  Tensor
fliplr :: Tensor -> Tensor
fliplr Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.fliplr_t) Tensor
_self

flipud ::
  -- | self
  Tensor ->
  Tensor
flipud :: Tensor -> Tensor
flipud Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.flipud_t) Tensor
_self

roll ::
  -- | self
  Tensor ->
  -- | shifts
  Int ->
  -- | dims
  Int ->
  Tensor
roll :: Tensor -> Int -> Int -> Tensor
roll Tensor
_self Int
_shifts Int
_dims = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> IO (ForeignPtr Tensor)
ATen.roll_tll) Tensor
_self Int
_shifts Int
_dims

rot90 ::
  -- | self
  Tensor ->
  -- | k
  Int ->
  -- | dims
  [Int] ->
  Tensor
rot90 :: Tensor -> Int -> [Int] -> Tensor
rot90 Tensor
_self Int
_k [Int]
_dims = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> Int64 -> ForeignPtr IntArray -> IO (ForeignPtr Tensor)
ATen.rot90_tll) Tensor
_self Int
_k [Int]
_dims

trapezoid_ttl ::
  -- | y
  Tensor ->
  -- | x
  Tensor ->
  -- | dim
  Int ->
  Tensor
trapezoid_ttl :: Tensor -> Tensor -> Int -> Tensor
trapezoid_ttl Tensor
_y Tensor
_x Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.trapezoid_ttl) Tensor
_y Tensor
_x Int
_dim

trapezoid_tsl ::
  -- | y
  Tensor ->
  -- | dx
  Float ->
  -- | dim
  Int ->
  Tensor
trapezoid_tsl :: Tensor -> Float -> Int -> Tensor
trapezoid_tsl Tensor
_y Float
_dx Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Scalar -> Int64 -> IO (ForeignPtr Tensor)
ATen.trapezoid_tsl) Tensor
_y Float
_dx Int
_dim

trapz ::
  -- | y
  Tensor ->
  -- | x
  Tensor ->
  -- | dim
  Int ->
  Tensor
trapz :: Tensor -> Tensor -> Int -> Tensor
trapz Tensor
_y Tensor
_x Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.trapz_ttl) Tensor
_y Tensor
_x Int
_dim

trapzScalar ::
  -- | y
  Tensor ->
  -- | dx
  Double ->
  -- | dim
  Int ->
  Tensor
trapzScalar :: Tensor -> Double -> Int -> Tensor
trapzScalar Tensor
_y Double
_dx Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> CDouble -> Int64 -> IO (ForeignPtr Tensor)
ATen.trapz_tdl) Tensor
_y Double
_dx Int
_dim

triplet_margin_loss ::
  -- | anchor
  Tensor ->
  -- | positive
  Tensor ->
  -- | negative
  Tensor ->
  -- | margin
  Double ->
  -- | p
  Double ->
  -- | eps
  Double ->
  -- | swap
  Bool ->
  -- | reduction
  Int ->
  Tensor
triplet_margin_loss :: Tensor
-> Tensor
-> Tensor
-> Double
-> Double
-> Double
-> Bool
-> Int
-> Tensor
triplet_margin_loss Tensor
_anchor Tensor
_positive Tensor
_negative Double
_margin Double
_p Double
_eps Bool
_swap Int
_reduction = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> IO y
cast8 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> CDouble
-> CDouble
-> CDouble
-> CBool
-> Int64
-> IO (ForeignPtr Tensor)
ATen.triplet_margin_loss_tttdddbl) Tensor
_anchor Tensor
_positive Tensor
_negative Double
_margin Double
_p Double
_eps Bool
_swap Int
_reduction

trunc ::
  -- | self
  Tensor ->
  Tensor
trunc :: Tensor -> Tensor
trunc Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.trunc_t) Tensor
_self

fix ::
  -- | self
  Tensor ->
  Tensor
fix :: Tensor -> Tensor
fix Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.fix_t) Tensor
_self

unique_dim ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | sorted
  Bool ->
  -- | return_inverse
  Bool ->
  -- | return_counts
  Bool ->
  (Tensor, Tensor, Tensor)
unique_dim :: Tensor -> Int -> Bool -> Bool -> Bool -> (Tensor, Tensor, Tensor)
unique_dim Tensor
_self Int
_dim Bool
_sorted Bool
_return_inverse Bool
_return_counts = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> Int64
-> CBool
-> CBool
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor, Tensor)))
ATen.unique_dim_tlbbb) Tensor
_self Int
_dim Bool
_sorted Bool
_return_inverse Bool
_return_counts

unique_consecutive ::
  -- | self
  Tensor ->
  -- | return_inverse
  Bool ->
  -- | return_counts
  Bool ->
  -- | dim
  Int ->
  (Tensor, Tensor, Tensor)
unique_consecutive :: Tensor -> Bool -> Bool -> Int -> (Tensor, Tensor, Tensor)
unique_consecutive Tensor
_self Bool
_return_inverse Bool
_return_counts Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> CBool
-> CBool
-> Int64
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor, Tensor)))
ATen.unique_consecutive_tbbl) Tensor
_self Bool
_return_inverse Bool
_return_counts Int
_dim

unique_dim_consecutive ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | return_inverse
  Bool ->
  -- | return_counts
  Bool ->
  (Tensor, Tensor, Tensor)
unique_dim_consecutive :: Tensor -> Int -> Bool -> Bool -> (Tensor, Tensor, Tensor)
unique_dim_consecutive Tensor
_self Int
_dim Bool
_return_inverse Bool
_return_counts = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64
-> CBool
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor, Tensor)))
ATen.unique_dim_consecutive_tlbb) Tensor
_self Int
_dim Bool
_return_inverse Bool
_return_counts

unsqueeze ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  Tensor
unsqueeze :: Tensor -> Int -> Tensor
unsqueeze Tensor
_self Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.unsqueeze_tl) Tensor
_self Int
_dim

vander ::
  -- | x
  Tensor ->
  -- | N
  Int ->
  -- | increasing
  Bool ->
  Tensor
vander :: Tensor -> Int -> Bool -> Tensor
vander Tensor
_x Int
_N Bool
_increasing = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> Int64 -> CBool -> IO (ForeignPtr Tensor)
ATen.vander_tlb) Tensor
_x Int
_N Bool
_increasing

var ::
  -- | self
  Tensor ->
  -- | unbiased
  Bool ->
  Tensor
var :: Tensor -> Bool -> Tensor
var Tensor
_self Bool
_unbiased = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> CBool -> IO (ForeignPtr Tensor)
ATen.var_tb) Tensor
_self Bool
_unbiased

varDim ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | unbiased
  Bool ->
  -- | keepdim
  Bool ->
  Tensor
varDim :: Tensor -> Int -> Bool -> Bool -> Tensor
varDim Tensor
_self Int
_dim Bool
_unbiased Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray -> CBool -> CBool -> IO (ForeignPtr Tensor)
ATen.var_tlbb) Tensor
_self Int
_dim Bool
_unbiased Bool
_keepdim

var_tllb ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | correction
  Int ->
  -- | keepdim
  Bool ->
  Tensor
var_tllb :: Tensor -> Int -> Int -> Bool -> Tensor
var_tllb Tensor
_self Int
_dim Int
_correction Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray -> Int64 -> CBool -> IO (ForeignPtr Tensor)
ATen.var_tllb) Tensor
_self Int
_dim Int
_correction Bool
_keepdim

varWithDimnames ::
  -- | self
  Tensor ->
  -- | dim
  [Dimname] ->
  -- | unbiased
  Bool ->
  -- | keepdim
  Bool ->
  Tensor
varWithDimnames :: Tensor -> [Dimname] -> Bool -> Bool -> Tensor
varWithDimnames Tensor
_self [Dimname]
_dim Bool
_unbiased Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr DimnameList
-> CBool
-> CBool
-> IO (ForeignPtr Tensor)
ATen.var_tNbb) Tensor
_self [Dimname]
_dim Bool
_unbiased Bool
_keepdim

var_tNlb ::
  -- | self
  Tensor ->
  -- | dim
  [Dimname] ->
  -- | correction
  Int ->
  -- | keepdim
  Bool ->
  Tensor
var_tNlb :: Tensor -> [Dimname] -> Int -> Bool -> Tensor
var_tNlb Tensor
_self [Dimname]
_dim Int
_correction Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr DimnameList
-> Int64
-> CBool
-> IO (ForeignPtr Tensor)
ATen.var_tNlb) Tensor
_self [Dimname]
_dim Int
_correction Bool
_keepdim

varMean ::
  -- | self
  Tensor ->
  -- | unbiased
  Bool ->
  (Tensor, Tensor)
varMean :: Tensor -> Bool -> (Tensor, Tensor)
varMean Tensor
_self Bool
_unbiased = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor
-> CBool -> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.var_mean_tb) Tensor
_self Bool
_unbiased

varMeanDim ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | unbiased
  Bool ->
  -- | keepdim
  Bool ->
  (Tensor, Tensor)
varMeanDim :: Tensor -> Int -> Bool -> Bool -> (Tensor, Tensor)
varMeanDim Tensor
_self Int
_dim Bool
_unbiased Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> CBool
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.var_mean_tlbb) Tensor
_self Int
_dim Bool
_unbiased Bool
_keepdim

var_mean_tllb ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | correction
  Int ->
  -- | keepdim
  Bool ->
  (Tensor, Tensor)
var_mean_tllb :: Tensor -> Int -> Int -> Bool -> (Tensor, Tensor)
var_mean_tllb Tensor
_self Int
_dim Int
_correction Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> Int64
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.var_mean_tllb) Tensor
_self Int
_dim Int
_correction Bool
_keepdim

varMeanWithDimnames ::
  -- | self
  Tensor ->
  -- | dim
  [Dimname] ->
  -- | unbiased
  Bool ->
  -- | keepdim
  Bool ->
  (Tensor, Tensor)
varMeanWithDimnames :: Tensor -> [Dimname] -> Bool -> Bool -> (Tensor, Tensor)
varMeanWithDimnames Tensor
_self [Dimname]
_dim Bool
_unbiased Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr DimnameList
-> CBool
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.var_mean_tNbb) Tensor
_self [Dimname]
_dim Bool
_unbiased Bool
_keepdim

var_mean_tNlb ::
  -- | self
  Tensor ->
  -- | dim
  [Dimname] ->
  -- | correction
  Int ->
  -- | keepdim
  Bool ->
  (Tensor, Tensor)
var_mean_tNlb :: Tensor -> [Dimname] -> Int -> Bool -> (Tensor, Tensor)
var_mean_tNlb Tensor
_self [Dimname]
_dim Int
_correction Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr DimnameList
-> Int64
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.var_mean_tNlb) Tensor
_self [Dimname]
_dim Int
_correction Bool
_keepdim

where' ::
  -- | condition
  Tensor ->
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
where' :: Tensor -> Tensor -> Tensor -> Tensor
where' Tensor
_condition Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.where_ttt) Tensor
_condition Tensor
_self Tensor
_other

where_tst ::
  -- | condition
  Tensor ->
  -- | self
  Float ->
  -- | other
  Tensor ->
  Tensor
where_tst :: Tensor -> Float -> Tensor -> Tensor
where_tst Tensor
_condition Float
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Scalar -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.where_tst) Tensor
_condition Float
_self Tensor
_other

where_tts ::
  -- | condition
  Tensor ->
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
where_tts :: Tensor -> Tensor -> Float -> Tensor
where_tts Tensor
_condition Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.where_tts) Tensor
_condition Tensor
_self Float
_other

where_tss ::
  -- | condition
  Tensor ->
  -- | self
  Float ->
  -- | other
  Float ->
  Tensor
where_tss :: Tensor -> Float -> Float -> Tensor
where_tss Tensor
_condition Float
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Scalar -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.where_tss) Tensor
_condition Float
_self Float
_other

isNonZero ::
  -- | condition
  Tensor ->
  [Tensor]
isNonZero :: Tensor -> [Tensor]
isNonZero Tensor
_condition = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr TensorList)
ATen.where_t) Tensor
_condition

norm_except_dim ::
  -- | v
  Tensor ->
  -- | pow
  Int ->
  -- | dim
  Int ->
  Tensor
norm_except_dim :: Tensor -> Int -> Int -> Tensor
norm_except_dim Tensor
_v Int
_pow Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> Int64 -> Int64 -> IO (ForeignPtr Tensor)
ATen.norm_except_dim_tll) Tensor
_v Int
_pow Int
_dim

native_norm_ts ::
  -- | self
  Tensor ->
  -- | p
  Float ->
  Tensor
native_norm_ts :: Tensor -> Float -> Tensor
native_norm_ts Tensor
_self Float
_p = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.native_norm_ts) Tensor
_self Float
_p

native_norm_tslbs ::
  -- | self
  Tensor ->
  -- | p
  Float ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  -- | dtype
  DType ->
  Tensor
native_norm_tslbs :: Tensor -> Float -> Int -> Bool -> DType -> Tensor
native_norm_tslbs Tensor
_self Float
_p Int
_dim Bool
_keepdim DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr IntArray
-> CBool
-> ScalarType
-> IO (ForeignPtr Tensor)
ATen.native_norm_tslbs) Tensor
_self Float
_p Int
_dim Bool
_keepdim DType
_dtype

normCastAll ::
  -- | self
  Tensor ->
  -- | p
  Float ->
  -- | dtype
  DType ->
  Tensor
normCastAll :: Tensor -> Float -> DType -> Tensor
normCastAll Tensor
_self Float
_p DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Scalar -> ScalarType -> IO (ForeignPtr Tensor)
ATen.norm_tss) Tensor
_self Float
_p DType
_dtype

normAll ::
  -- | self
  Tensor ->
  -- | p
  Float ->
  Tensor
normAll :: Tensor -> Float -> Tensor
normAll Tensor
_self Float
_p = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.norm_ts) Tensor
_self Float
_p

normCastDim ::
  -- | self
  Tensor ->
  -- | p
  Float ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  -- | dtype
  DType ->
  Tensor
normCastDim :: Tensor -> Float -> Int -> Bool -> DType -> Tensor
normCastDim Tensor
_self Float
_p Int
_dim Bool
_keepdim DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr IntArray
-> CBool
-> ScalarType
-> IO (ForeignPtr Tensor)
ATen.norm_tslbs) Tensor
_self Float
_p Int
_dim Bool
_keepdim DType
_dtype

normDim ::
  -- | self
  Tensor ->
  -- | p
  Float ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  Tensor
normDim :: Tensor -> Float -> Int -> Bool -> Tensor
normDim Tensor
_self Float
_p Int
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr IntArray
-> CBool
-> IO (ForeignPtr Tensor)
ATen.norm_tslb) Tensor
_self Float
_p Int
_dim Bool
_keepdim

norm_tsNbs ::
  -- | self
  Tensor ->
  -- | p
  Float ->
  -- | dim
  [Dimname] ->
  -- | keepdim
  Bool ->
  -- | dtype
  DType ->
  Tensor
norm_tsNbs :: Tensor -> Float -> [Dimname] -> Bool -> DType -> Tensor
norm_tsNbs Tensor
_self Float
_p [Dimname]
_dim Bool
_keepdim DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr DimnameList
-> CBool
-> ScalarType
-> IO (ForeignPtr Tensor)
ATen.norm_tsNbs) Tensor
_self Float
_p [Dimname]
_dim Bool
_keepdim DType
_dtype

norm_tsNb ::
  -- | self
  Tensor ->
  -- | p
  Float ->
  -- | dim
  [Dimname] ->
  -- | keepdim
  Bool ->
  Tensor
norm_tsNb :: Tensor -> Float -> [Dimname] -> Bool -> Tensor
norm_tsNb Tensor
_self Float
_p [Dimname]
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr DimnameList
-> CBool
-> IO (ForeignPtr Tensor)
ATen.norm_tsNb) Tensor
_self Float
_p [Dimname]
_dim Bool
_keepdim

frexp ::
  -- | self
  Tensor ->
  (Tensor, Tensor)
frexp :: Tensor -> (Tensor, Tensor)
frexp Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.frexp_t) Tensor
_self

frobeniusNormAll ::
  -- | self
  Tensor ->
  Tensor
frobeniusNormAll :: Tensor -> Tensor
frobeniusNormAll Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.frobenius_norm_t) Tensor
_self

frobeniusNormDim ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  Tensor
frobeniusNormDim :: Tensor -> Int -> Bool -> Tensor
frobeniusNormDim Tensor
_self Int
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr IntArray -> CBool -> IO (ForeignPtr Tensor)
ATen.frobenius_norm_tlb) Tensor
_self Int
_dim Bool
_keepdim

nuclearNormAll ::
  -- | self
  Tensor ->
  -- | keepdim
  Bool ->
  Tensor
nuclearNormAll :: Tensor -> Bool -> Tensor
nuclearNormAll Tensor
_self Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> CBool -> IO (ForeignPtr Tensor)
ATen.nuclear_norm_tb) Tensor
_self Bool
_keepdim

nuclearNormDim ::
  -- | self
  Tensor ->
  -- | dim
  (Int, Int) ->
  -- | keepdim
  Bool ->
  Tensor
nuclearNormDim :: Tensor -> (Int, Int) -> Bool -> Tensor
nuclearNormDim Tensor
_self (Int, Int)
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr IntArray -> CBool -> IO (ForeignPtr Tensor)
ATen.nuclear_norm_tlb) Tensor
_self (Int, Int)
_dim Bool
_keepdim

clone ::
  -- | self
  Tensor ->
  -- | memory_format
  ATen.MemoryFormat ->
  Tensor
clone :: Tensor -> ScalarType -> Tensor
clone Tensor
_self ScalarType
_memory_format = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ScalarType -> IO (ForeignPtr Tensor)
ATen.clone_tM) Tensor
_self ScalarType
_memory_format

positive ::
  -- | self
  Tensor ->
  Tensor
positive :: Tensor -> Tensor
positive Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.positive_t) Tensor
_self

sub ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  -- | alpha
  Float ->
  Tensor
sub :: Tensor -> Tensor -> Float -> Tensor
sub Tensor
_self Tensor
_other Float
_alpha = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.sub_tts) Tensor
_self Tensor
_other Float
_alpha

subScalar ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  -- | alpha
  Float ->
  Tensor
subScalar :: Tensor -> Float -> Float -> Tensor
subScalar Tensor
_self Float
_other Float
_alpha = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Scalar -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.sub_tss) Tensor
_self Float
_other Float
_alpha

subtract_tts ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  -- | alpha
  Float ->
  Tensor
subtract_tts :: Tensor -> Tensor -> Float -> Tensor
subtract_tts Tensor
_self Tensor
_other Float
_alpha = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.subtract_tts) Tensor
_self Tensor
_other Float
_alpha

subtract_tss ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  -- | alpha
  Float ->
  Tensor
subtract_tss :: Tensor -> Float -> Float -> Tensor
subtract_tss Tensor
_self Float
_other Float
_alpha = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Scalar -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.subtract_tss) Tensor
_self Float
_other Float
_alpha

rsub ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  -- | alpha
  Float ->
  Tensor
rsub :: Tensor -> Tensor -> Float -> Tensor
rsub Tensor
_self Tensor
_other Float
_alpha = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.rsub_tts) Tensor
_self Tensor
_other Float
_alpha

heaviside ::
  -- | self
  Tensor ->
  -- | values
  Tensor ->
  Tensor
heaviside :: Tensor -> Tensor -> Tensor
heaviside Tensor
_self Tensor
_values = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.heaviside_tt) Tensor
_self Tensor
_values

rsubScalar ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  -- | alpha
  Float ->
  Tensor
rsubScalar :: Tensor -> Float -> Float -> Tensor
rsubScalar Tensor
_self Float
_other Float
_alpha = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Scalar -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.rsub_tss) Tensor
_self Float
_other Float
_alpha

sparse_sampled_addmm ::
  -- | self
  Tensor ->
  -- | mat1
  Tensor ->
  -- | mat2
  Tensor ->
  -- | beta
  Float ->
  -- | alpha
  Float ->
  Tensor
sparse_sampled_addmm :: Tensor -> Tensor -> Tensor -> Float -> Float -> Tensor
sparse_sampled_addmm Tensor
_self Tensor
_mat1 Tensor
_mat2 Float
_beta Float
_alpha = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr Scalar
-> IO (ForeignPtr Tensor)
ATen.sparse_sampled_addmm_tttss) Tensor
_self Tensor
_mat1 Tensor
_mat2 Float
_beta Float
_alpha

addmm ::
  -- | self
  Tensor ->
  -- | mat1
  Tensor ->
  -- | mat2
  Tensor ->
  -- | beta
  Float ->
  -- | alpha
  Float ->
  Tensor
addmm :: Tensor -> Tensor -> Tensor -> Float -> Float -> Tensor
addmm Tensor
_self Tensor
_mat1 Tensor
_mat2 Float
_beta Float
_alpha = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr Scalar
-> IO (ForeignPtr Tensor)
ATen.addmm_tttss) Tensor
_self Tensor
_mat1 Tensor
_mat2 Float
_beta Float
_alpha

hspmm ::
  -- | mat1
  Tensor ->
  -- | mat2
  Tensor ->
  Tensor
hspmm :: Tensor -> Tensor -> Tensor
hspmm Tensor
_mat1 Tensor
_mat2 = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.hspmm_tt) Tensor
_mat1 Tensor
_mat2

unbind ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  [Tensor]
unbind :: Tensor -> Int -> [Tensor]
unbind Tensor
_self Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO (ForeignPtr TensorList)
ATen.unbind_tl) Tensor
_self Int
_dim

unbindWithDimname ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  [Tensor]
unbindWithDimname :: Tensor -> Dimname -> [Tensor]
unbindWithDimname Tensor
_self Dimname
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor
-> ForeignPtr Dimname -> IO (ForeignPtr TensorList)
ATen.unbind_tn) Tensor
_self Dimname
_dim

mkldnn_reorder_conv2d_weight ::
  -- | self
  Tensor ->
  -- | padding
  (Int, Int) ->
  -- | stride
  (Int, Int) ->
  -- | dilation
  (Int, Int) ->
  -- | groups
  Int ->
  Tensor
mkldnn_reorder_conv2d_weight :: Tensor -> (Int, Int) -> (Int, Int) -> (Int, Int) -> Int -> Tensor
mkldnn_reorder_conv2d_weight Tensor
_self (Int, Int)
_padding (Int, Int)
_stride (Int, Int)
_dilation Int
_groups = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> Int64
-> IO (ForeignPtr Tensor)
ATen.mkldnn_reorder_conv2d_weight_tllll) Tensor
_self (Int, Int)
_padding (Int, Int)
_stride (Int, Int)
_dilation Int
_groups

mkldnn_reorder_conv3d_weight ::
  -- | self
  Tensor ->
  -- | padding
  (Int, Int, Int) ->
  -- | stride
  (Int, Int, Int) ->
  -- | dilation
  (Int, Int, Int) ->
  -- | groups
  Int ->
  Tensor
mkldnn_reorder_conv3d_weight :: Tensor
-> (Int, Int, Int)
-> (Int, Int, Int)
-> (Int, Int, Int)
-> Int
-> Tensor
mkldnn_reorder_conv3d_weight Tensor
_self (Int, Int, Int)
_padding (Int, Int, Int)
_stride (Int, Int, Int)
_dilation Int
_groups = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> Int64
-> IO (ForeignPtr Tensor)
ATen.mkldnn_reorder_conv3d_weight_tllll) Tensor
_self (Int, Int, Int)
_padding (Int, Int, Int)
_stride (Int, Int, Int)
_dilation Int
_groups

quantize_per_tensor_dynamic ::
  -- | self
  Tensor ->
  -- | dtype
  DType ->
  -- | reduce_range
  Bool ->
  Tensor
quantize_per_tensor_dynamic :: Tensor -> DType -> Bool -> Tensor
quantize_per_tensor_dynamic Tensor
_self DType
_dtype Bool
_reduce_range = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> ScalarType -> CBool -> IO (ForeignPtr Tensor)
ATen.quantize_per_tensor_dynamic_tsb) Tensor
_self DType
_dtype Bool
_reduce_range

quantize_per_tensor_tdls ::
  -- | self
  Tensor ->
  -- | scale
  Double ->
  -- | zero_point
  Int ->
  -- | dtype
  DType ->
  Tensor
quantize_per_tensor_tdls :: Tensor -> Double -> Int -> DType -> Tensor
quantize_per_tensor_tdls Tensor
_self Double
_scale Int
_zero_point DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> CDouble -> Int64 -> ScalarType -> IO (ForeignPtr Tensor)
ATen.quantize_per_tensor_tdls) Tensor
_self Double
_scale Int
_zero_point DType
_dtype

quantize_per_tensor_ttts ::
  -- | self
  Tensor ->
  -- | scale
  Tensor ->
  -- | zero_point
  Tensor ->
  -- | dtype
  DType ->
  Tensor
quantize_per_tensor_ttts :: Tensor -> Tensor -> Tensor -> DType -> Tensor
quantize_per_tensor_ttts Tensor
_self Tensor
_scale Tensor
_zero_point DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ScalarType
-> IO (ForeignPtr Tensor)
ATen.quantize_per_tensor_ttts) Tensor
_self Tensor
_scale Tensor
_zero_point DType
_dtype

quantize_per_tensor_ltts ::
  -- | tensors
  [Tensor] ->
  -- | scales
  Tensor ->
  -- | zero_points
  Tensor ->
  -- | dtype
  DType ->
  [Tensor]
quantize_per_tensor_ltts :: [Tensor] -> Tensor -> Tensor -> DType -> [Tensor]
quantize_per_tensor_ltts [Tensor]
_tensors Tensor
_scales Tensor
_zero_points DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr TensorList
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ScalarType
-> IO (ForeignPtr TensorList)
ATen.quantize_per_tensor_ltts) [Tensor]
_tensors Tensor
_scales Tensor
_zero_points DType
_dtype

quantize_per_channel ::
  -- | self
  Tensor ->
  -- | scales
  Tensor ->
  -- | zero_points
  Tensor ->
  -- | axis
  Int ->
  -- | dtype
  DType ->
  Tensor
quantize_per_channel :: Tensor -> Tensor -> Tensor -> Int -> DType -> Tensor
quantize_per_channel Tensor
_self Tensor
_scales Tensor
_zero_points Int
_axis DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> Int64
-> ScalarType
-> IO (ForeignPtr Tensor)
ATen.quantize_per_channel_tttls) Tensor
_self Tensor
_scales Tensor
_zero_points Int
_axis DType
_dtype

dequantize_t ::
  -- | self
  Tensor ->
  Tensor
dequantize_t :: Tensor -> Tensor
dequantize_t Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.dequantize_t) Tensor
_self

dequantize_l ::
  -- | tensors
  [Tensor] ->
  [Tensor]
dequantize_l :: [Tensor] -> [Tensor]
dequantize_l [Tensor]
_tensors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr TensorList -> IO (ForeignPtr TensorList)
ATen.dequantize_l) [Tensor]
_tensors

q_scale ::
  -- | self
  Tensor ->
  Double
q_scale :: Tensor -> Double
q_scale Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO CDouble
ATen.q_scale_t) Tensor
_self

q_zero_point ::
  -- | self
  Tensor ->
  Int
q_zero_point :: Tensor -> Int
q_zero_point Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO Int64
ATen.q_zero_point_t) Tensor
_self

q_per_channel_scales ::
  -- | self
  Tensor ->
  Tensor
q_per_channel_scales :: Tensor -> Tensor
q_per_channel_scales Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.q_per_channel_scales_t) Tensor
_self

q_per_channel_zero_points ::
  -- | self
  Tensor ->
  Tensor
q_per_channel_zero_points :: Tensor -> Tensor
q_per_channel_zero_points Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.q_per_channel_zero_points_t) Tensor
_self

q_per_channel_axis ::
  -- | self
  Tensor ->
  Int
q_per_channel_axis :: Tensor -> Int
q_per_channel_axis Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO Int64
ATen.q_per_channel_axis_t) Tensor
_self

int_repr ::
  -- | self
  Tensor ->
  Tensor
int_repr :: Tensor -> Tensor
int_repr Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.int_repr_t) Tensor
_self

fake_quantize_per_tensor_affine_tdlll ::
  -- | self
  Tensor ->
  -- | scale
  Double ->
  -- | zero_point
  Int ->
  -- | quant_min
  Int ->
  -- | quant_max
  Int ->
  Tensor
fake_quantize_per_tensor_affine_tdlll :: Tensor -> Double -> Int -> Int -> Int -> Tensor
fake_quantize_per_tensor_affine_tdlll Tensor
_self Double
_scale Int
_zero_point Int
_quant_min Int
_quant_max = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> CDouble -> Int64 -> Int64 -> Int64 -> IO (ForeignPtr Tensor)
ATen.fake_quantize_per_tensor_affine_tdlll) Tensor
_self Double
_scale Int
_zero_point Int
_quant_min Int
_quant_max

fake_quantize_per_tensor_affine_tttll ::
  -- | self
  Tensor ->
  -- | scale
  Tensor ->
  -- | zero_point
  Tensor ->
  -- | quant_min
  Int ->
  -- | quant_max
  Int ->
  Tensor
fake_quantize_per_tensor_affine_tttll :: Tensor -> Tensor -> Tensor -> Int -> Int -> Tensor
fake_quantize_per_tensor_affine_tttll Tensor
_self Tensor
_scale Tensor
_zero_point Int
_quant_min Int
_quant_max = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> Int64
-> Int64
-> IO (ForeignPtr Tensor)
ATen.fake_quantize_per_tensor_affine_tttll) Tensor
_self Tensor
_scale Tensor
_zero_point Int
_quant_min Int
_quant_max

fake_quantize_per_tensor_affine_cachemask ::
  -- | self
  Tensor ->
  -- | scale
  Double ->
  -- | zero_point
  Int ->
  -- | quant_min
  Int ->
  -- | quant_max
  Int ->
  (Tensor, Tensor)
fake_quantize_per_tensor_affine_cachemask :: Tensor -> Double -> Int -> Int -> Int -> (Tensor, Tensor)
fake_quantize_per_tensor_affine_cachemask Tensor
_self Double
_scale Int
_zero_point Int
_quant_min Int
_quant_max = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> CDouble
-> Int64
-> Int64
-> Int64
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.fake_quantize_per_tensor_affine_cachemask_tdlll) Tensor
_self Double
_scale Int
_zero_point Int
_quant_min Int
_quant_max

fake_quantize_per_channel_affine ::
  -- | self
  Tensor ->
  -- | scale
  Tensor ->
  -- | zero_point
  Tensor ->
  -- | axis
  Int ->
  -- | quant_min
  Int ->
  -- | quant_max
  Int ->
  Tensor
fake_quantize_per_channel_affine :: Tensor -> Tensor -> Tensor -> Int -> Int -> Int -> Tensor
fake_quantize_per_channel_affine Tensor
_self Tensor
_scale Tensor
_zero_point Int
_axis Int
_quant_min Int
_quant_max = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> Int64
-> Int64
-> Int64
-> IO (ForeignPtr Tensor)
ATen.fake_quantize_per_channel_affine_tttlll) Tensor
_self Tensor
_scale Tensor
_zero_point Int
_axis Int
_quant_min Int
_quant_max

fake_quantize_per_channel_affine_cachemask ::
  -- | self
  Tensor ->
  -- | scale
  Tensor ->
  -- | zero_point
  Tensor ->
  -- | axis
  Int ->
  -- | quant_min
  Int ->
  -- | quant_max
  Int ->
  (Tensor, Tensor)
fake_quantize_per_channel_affine_cachemask :: Tensor -> Tensor -> Tensor -> Int -> Int -> Int -> (Tensor, Tensor)
fake_quantize_per_channel_affine_cachemask Tensor
_self Tensor
_scale Tensor
_zero_point Int
_axis Int
_quant_min Int
_quant_max = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> Int64
-> Int64
-> Int64
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.fake_quantize_per_channel_affine_cachemask_tttlll) Tensor
_self Tensor
_scale Tensor
_zero_point Int
_axis Int
_quant_min Int
_quant_max

fused_moving_avg_obs_fake_quant ::
  -- | self
  Tensor ->
  -- | observer_on
  Tensor ->
  -- | fake_quant_on
  Tensor ->
  -- | running_min
  Tensor ->
  -- | running_max
  Tensor ->
  -- | scale
  Tensor ->
  -- | zero_point
  Tensor ->
  -- | averaging_const
  Double ->
  -- | quant_min
  Int ->
  -- | quant_max
  Int ->
  -- | ch_axis
  Int ->
  -- | per_row_fake_quant
  Bool ->
  -- | symmetric_quant
  Bool ->
  Tensor
fused_moving_avg_obs_fake_quant :: Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Double
-> Int
-> Int
-> Int
-> Bool
-> Bool
-> Tensor
fused_moving_avg_obs_fake_quant Tensor
_self Tensor
_observer_on Tensor
_fake_quant_on Tensor
_running_min Tensor
_running_max Tensor
_scale Tensor
_zero_point Double
_averaging_const Int
_quant_min Int
_quant_max Int
_ch_axis Bool
_per_row_fake_quant Bool
_symmetric_quant = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       x9 cx9 x10 cx10 x11 cx11 x12 cx12 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable x9 cx9, Castable x10 cx10,
 Castable x11 cx11, Castable x12 cx12, Castable y cy) =>
(ca
 -> cx1
 -> cx2
 -> cx3
 -> cx4
 -> cx5
 -> cx6
 -> cx7
 -> cx8
 -> cx9
 -> cx10
 -> cx11
 -> cx12
 -> IO cy)
-> a
-> x1
-> x2
-> x3
-> x4
-> x5
-> x6
-> x7
-> x8
-> x9
-> x10
-> x11
-> x12
-> IO y
cast13 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> CDouble
-> Int64
-> Int64
-> Int64
-> CBool
-> CBool
-> IO (ForeignPtr Tensor)
ATen.fused_moving_avg_obs_fake_quant_tttttttdlllbb) Tensor
_self Tensor
_observer_on Tensor
_fake_quant_on Tensor
_running_min Tensor
_running_max Tensor
_scale Tensor
_zero_point Double
_averaging_const Int
_quant_min Int
_quant_max Int
_ch_axis Bool
_per_row_fake_quant Bool
_symmetric_quant

choose_qparams_optimized ::
  -- | input
  Tensor ->
  -- | numel
  Int ->
  -- | n_bins
  Int ->
  -- | ratio
  Double ->
  -- | bit_width
  Int ->
  (Tensor, Tensor)
choose_qparams_optimized :: Tensor -> Int -> Int -> Double -> Int -> (Tensor, Tensor)
choose_qparams_optimized Tensor
_input Int
_numel Int
_n_bins Double
_ratio Int
_bit_width = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> Int64
-> Int64
-> CDouble
-> Int64
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.choose_qparams_optimized_tlldl) Tensor
_input Int
_numel Int
_n_bins Double
_ratio Int
_bit_width

meshgrid_l ::
  -- | tensors
  [Tensor] ->
  [Tensor]
meshgrid_l :: [Tensor] -> [Tensor]
meshgrid_l [Tensor]
_tensors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr TensorList -> IO (ForeignPtr TensorList)
ATen.meshgrid_l) [Tensor]
_tensors

meshgrid_ls ::
  -- | tensors
  [Tensor] ->
  -- | indexing
  String ->
  [Tensor]
meshgrid_ls :: [Tensor] -> String -> [Tensor]
meshgrid_ls [Tensor]
_tensors String
_indexing = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr TensorList
-> ForeignPtr StdString -> IO (ForeignPtr TensorList)
ATen.meshgrid_ls) [Tensor]
_tensors String
_indexing

cartesian_prod ::
  -- | tensors
  [Tensor] ->
  Tensor
cartesian_prod :: [Tensor] -> Tensor
cartesian_prod [Tensor]
_tensors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr TensorList -> IO (ForeignPtr Tensor)
ATen.cartesian_prod_l) [Tensor]
_tensors

combinations ::
  -- | self
  Tensor ->
  -- | r
  Int ->
  -- | with_replacement
  Bool ->
  Tensor
combinations :: Tensor -> Int -> Bool -> Tensor
combinations Tensor
_self Int
_r Bool
_with_replacement = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> Int64 -> CBool -> IO (ForeignPtr Tensor)
ATen.combinations_tlb) Tensor
_self Int
_r Bool
_with_replacement

resultType ::
  -- | tensor
  Tensor ->
  -- | other
  Tensor ->
  DType
resultType :: Tensor -> Tensor -> DType
resultType Tensor
_tensor Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO ScalarType
ATen.result_type_tt) Tensor
_tensor Tensor
_other

resultTypeScalar ::
  -- | tensor
  Tensor ->
  -- | other
  Float ->
  DType
resultTypeScalar :: Tensor -> Float -> DType
resultTypeScalar Tensor
_tensor Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO ScalarType
ATen.result_type_ts) Tensor
_tensor Float
_other

resultTypeScalar' ::
  -- | scalar
  Float ->
  -- | tensor
  Tensor ->
  DType
resultTypeScalar' :: Float -> Tensor -> DType
resultTypeScalar' Float
_scalar Tensor
_tensor = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Scalar -> ForeignPtr Tensor -> IO ScalarType
ATen.result_type_st) Float
_scalar Tensor
_tensor

resultTypeScalars ::
  -- | scalar1
  Float ->
  -- | scalar2
  Float ->
  DType
resultTypeScalars :: Float -> Float -> DType
resultTypeScalars Float
_scalar1 Float
_scalar2 = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Scalar -> ForeignPtr Scalar -> IO ScalarType
ATen.result_type_ss) Float
_scalar1 Float
_scalar2

can_cast ::
  -- | from
  DType ->
  -- | to
  DType ->
  Bool
can_cast :: DType -> DType -> Bool
can_cast DType
_from DType
_to = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ScalarType -> ScalarType -> IO CBool
ATen.can_cast_ss) DType
_from DType
_to

promote_types ::
  -- | type1
  DType ->
  -- | type2
  DType ->
  DType
promote_types :: DType -> DType -> DType
promote_types DType
_type1 DType
_type2 = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ScalarType -> ScalarType -> IO ScalarType
ATen.promote_types_ss) DType
_type1 DType
_type2

lstm ::
  -- | input
  Tensor ->
  -- | hx
  [Tensor] ->
  -- | params
  [Tensor] ->
  -- | has_biases
  Bool ->
  -- | num_layers
  Int ->
  -- | dropout
  Double ->
  -- | train
  Bool ->
  -- | bidirectional
  Bool ->
  -- | batch_first
  Bool ->
  (Tensor, Tensor, Tensor)
lstm :: Tensor
-> [Tensor]
-> [Tensor]
-> Bool
-> Int
-> Double
-> Bool
-> Bool
-> Bool
-> (Tensor, Tensor, Tensor)
lstm Tensor
_input [Tensor]
_hx [Tensor]
_params Bool
_has_biases Int
_num_layers Double
_dropout Bool
_train Bool
_bidirectional Bool
_batch_first = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable y cy) =>
(ca
 -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> cx8 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> x8 -> IO y
cast9 ForeignPtr Tensor
-> ForeignPtr TensorList
-> ForeignPtr TensorList
-> CBool
-> Int64
-> CDouble
-> CBool
-> CBool
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor, Tensor)))
ATen.lstm_tllbldbbb) Tensor
_input [Tensor]
_hx [Tensor]
_params Bool
_has_biases Int
_num_layers Double
_dropout Bool
_train Bool
_bidirectional Bool
_batch_first

lstm' ::
  -- | data
  Tensor ->
  -- | batch_sizes
  Tensor ->
  -- | hx
  [Tensor] ->
  -- | params
  [Tensor] ->
  -- | has_biases
  Bool ->
  -- | num_layers
  Int ->
  -- | dropout
  Double ->
  -- | train
  Bool ->
  -- | bidirectional
  Bool ->
  (Tensor, Tensor, Tensor)
lstm' :: Tensor
-> Tensor
-> [Tensor]
-> [Tensor]
-> Bool
-> Int
-> Double
-> Bool
-> Bool
-> (Tensor, Tensor, Tensor)
lstm' Tensor
_data Tensor
_batch_sizes [Tensor]
_hx [Tensor]
_params Bool
_has_biases Int
_num_layers Double
_dropout Bool
_train Bool
_bidirectional = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable y cy) =>
(ca
 -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> cx8 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> x8 -> IO y
cast9 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr TensorList
-> ForeignPtr TensorList
-> CBool
-> Int64
-> CDouble
-> CBool
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor, Tensor)))
ATen.lstm_ttllbldbb) Tensor
_data Tensor
_batch_sizes [Tensor]
_hx [Tensor]
_params Bool
_has_biases Int
_num_layers Double
_dropout Bool
_train Bool
_bidirectional

gru ::
  -- | input
  Tensor ->
  -- | hx
  Tensor ->
  -- | params
  [Tensor] ->
  -- | has_biases
  Bool ->
  -- | num_layers
  Int ->
  -- | dropout
  Double ->
  -- | train
  Bool ->
  -- | bidirectional
  Bool ->
  -- | batch_first
  Bool ->
  (Tensor, Tensor)
gru :: Tensor
-> Tensor
-> [Tensor]
-> Bool
-> Int
-> Double
-> Bool
-> Bool
-> Bool
-> (Tensor, Tensor)
gru Tensor
_input Tensor
_hx [Tensor]
_params Bool
_has_biases Int
_num_layers Double
_dropout Bool
_train Bool
_bidirectional Bool
_batch_first = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable y cy) =>
(ca
 -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> cx8 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> x8 -> IO y
cast9 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr TensorList
-> CBool
-> Int64
-> CDouble
-> CBool
-> CBool
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.gru_ttlbldbbb) Tensor
_input Tensor
_hx [Tensor]
_params Bool
_has_biases Int
_num_layers Double
_dropout Bool
_train Bool
_bidirectional Bool
_batch_first

gru' ::
  -- | data
  Tensor ->
  -- | batch_sizes
  Tensor ->
  -- | hx
  Tensor ->
  -- | params
  [Tensor] ->
  -- | has_biases
  Bool ->
  -- | num_layers
  Int ->
  -- | dropout
  Double ->
  -- | train
  Bool ->
  -- | bidirectional
  Bool ->
  (Tensor, Tensor)
gru' :: Tensor
-> Tensor
-> Tensor
-> [Tensor]
-> Bool
-> Int
-> Double
-> Bool
-> Bool
-> (Tensor, Tensor)
gru' Tensor
_data Tensor
_batch_sizes Tensor
_hx [Tensor]
_params Bool
_has_biases Int
_num_layers Double
_dropout Bool
_train Bool
_bidirectional = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable y cy) =>
(ca
 -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> cx8 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> x8 -> IO y
cast9 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr TensorList
-> CBool
-> Int64
-> CDouble
-> CBool
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.gru_tttlbldbb) Tensor
_data Tensor
_batch_sizes Tensor
_hx [Tensor]
_params Bool
_has_biases Int
_num_layers Double
_dropout Bool
_train Bool
_bidirectional

rnnTanh ::
  -- | input
  Tensor ->
  -- | hx
  Tensor ->
  -- | params
  [Tensor] ->
  -- | has_biases
  Bool ->
  -- | num_layers
  Int ->
  -- | dropout
  Double ->
  -- | train
  Bool ->
  -- | bidirectional
  Bool ->
  -- | batch_first
  Bool ->
  (Tensor, Tensor)
rnnTanh :: Tensor
-> Tensor
-> [Tensor]
-> Bool
-> Int
-> Double
-> Bool
-> Bool
-> Bool
-> (Tensor, Tensor)
rnnTanh Tensor
_input Tensor
_hx [Tensor]
_params Bool
_has_biases Int
_num_layers Double
_dropout Bool
_train Bool
_bidirectional Bool
_batch_first = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable y cy) =>
(ca
 -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> cx8 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> x8 -> IO y
cast9 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr TensorList
-> CBool
-> Int64
-> CDouble
-> CBool
-> CBool
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.rnn_tanh_ttlbldbbb) Tensor
_input Tensor
_hx [Tensor]
_params Bool
_has_biases Int
_num_layers Double
_dropout Bool
_train Bool
_bidirectional Bool
_batch_first

rnnTanh' ::
  -- | data
  Tensor ->
  -- | batch_sizes
  Tensor ->
  -- | hx
  Tensor ->
  -- | params
  [Tensor] ->
  -- | has_biases
  Bool ->
  -- | num_layers
  Int ->
  -- | dropout
  Double ->
  -- | train
  Bool ->
  -- | bidirectional
  Bool ->
  (Tensor, Tensor)
rnnTanh' :: Tensor
-> Tensor
-> Tensor
-> [Tensor]
-> Bool
-> Int
-> Double
-> Bool
-> Bool
-> (Tensor, Tensor)
rnnTanh' Tensor
_data Tensor
_batch_sizes Tensor
_hx [Tensor]
_params Bool
_has_biases Int
_num_layers Double
_dropout Bool
_train Bool
_bidirectional = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable y cy) =>
(ca
 -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> cx8 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> x8 -> IO y
cast9 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr TensorList
-> CBool
-> Int64
-> CDouble
-> CBool
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.rnn_tanh_tttlbldbb) Tensor
_data Tensor
_batch_sizes Tensor
_hx [Tensor]
_params Bool
_has_biases Int
_num_layers Double
_dropout Bool
_train Bool
_bidirectional

rnnRelu ::
  -- | input
  Tensor ->
  -- | hx
  Tensor ->
  -- | params
  [Tensor] ->
  -- | has_biases
  Bool ->
  -- | num_layers
  Int ->
  -- | dropout
  Double ->
  -- | train
  Bool ->
  -- | bidirectional
  Bool ->
  -- | batch_first
  Bool ->
  (Tensor, Tensor)
rnnRelu :: Tensor
-> Tensor
-> [Tensor]
-> Bool
-> Int
-> Double
-> Bool
-> Bool
-> Bool
-> (Tensor, Tensor)
rnnRelu Tensor
_input Tensor
_hx [Tensor]
_params Bool
_has_biases Int
_num_layers Double
_dropout Bool
_train Bool
_bidirectional Bool
_batch_first = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable y cy) =>
(ca
 -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> cx8 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> x8 -> IO y
cast9 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr TensorList
-> CBool
-> Int64
-> CDouble
-> CBool
-> CBool
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.rnn_relu_ttlbldbbb) Tensor
_input Tensor
_hx [Tensor]
_params Bool
_has_biases Int
_num_layers Double
_dropout Bool
_train Bool
_bidirectional Bool
_batch_first

rnnRelu' ::
  -- | data
  Tensor ->
  -- | batch_sizes
  Tensor ->
  -- | hx
  Tensor ->
  -- | params
  [Tensor] ->
  -- | has_biases
  Bool ->
  -- | num_layers
  Int ->
  -- | dropout
  Double ->
  -- | train
  Bool ->
  -- | bidirectional
  Bool ->
  (Tensor, Tensor)
rnnRelu' :: Tensor
-> Tensor
-> Tensor
-> [Tensor]
-> Bool
-> Int
-> Double
-> Bool
-> Bool
-> (Tensor, Tensor)
rnnRelu' Tensor
_data Tensor
_batch_sizes Tensor
_hx [Tensor]
_params Bool
_has_biases Int
_num_layers Double
_dropout Bool
_train Bool
_bidirectional = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable y cy) =>
(ca
 -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> cx8 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> x8 -> IO y
cast9 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr TensorList
-> CBool
-> Int64
-> CDouble
-> CBool
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.rnn_relu_tttlbldbb) Tensor
_data Tensor
_batch_sizes Tensor
_hx [Tensor]
_params Bool
_has_biases Int
_num_layers Double
_dropout Bool
_train Bool
_bidirectional

lstm_cell ::
  -- | input
  Tensor ->
  -- | hx
  [Tensor] ->
  -- | w_ih
  Tensor ->
  -- | w_hh
  Tensor ->
  -- | b_ih
  Tensor ->
  -- | b_hh
  Tensor ->
  (Tensor, Tensor)
lstm_cell :: Tensor
-> [Tensor]
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> (Tensor, Tensor)
lstm_cell Tensor
_input [Tensor]
_hx Tensor
_w_ih Tensor
_w_hh Tensor
_b_ih Tensor
_b_hh = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr TensorList
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.lstm_cell_tltttt) Tensor
_input [Tensor]
_hx Tensor
_w_ih Tensor
_w_hh Tensor
_b_ih Tensor
_b_hh

gru_cell ::
  -- | input
  Tensor ->
  -- | hx
  Tensor ->
  -- | w_ih
  Tensor ->
  -- | w_hh
  Tensor ->
  -- | b_ih
  Tensor ->
  -- | b_hh
  Tensor ->
  Tensor
gru_cell :: Tensor -> Tensor -> Tensor -> Tensor -> Tensor -> Tensor -> Tensor
gru_cell Tensor
_input Tensor
_hx Tensor
_w_ih Tensor
_w_hh Tensor
_b_ih Tensor
_b_hh = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> IO (ForeignPtr Tensor)
ATen.gru_cell_tttttt) Tensor
_input Tensor
_hx Tensor
_w_ih Tensor
_w_hh Tensor
_b_ih Tensor
_b_hh

rnn_tanh_cell ::
  -- | input
  Tensor ->
  -- | hx
  Tensor ->
  -- | w_ih
  Tensor ->
  -- | w_hh
  Tensor ->
  -- | b_ih
  Tensor ->
  -- | b_hh
  Tensor ->
  Tensor
rnn_tanh_cell :: Tensor -> Tensor -> Tensor -> Tensor -> Tensor -> Tensor -> Tensor
rnn_tanh_cell Tensor
_input Tensor
_hx Tensor
_w_ih Tensor
_w_hh Tensor
_b_ih Tensor
_b_hh = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> IO (ForeignPtr Tensor)
ATen.rnn_tanh_cell_tttttt) Tensor
_input Tensor
_hx Tensor
_w_ih Tensor
_w_hh Tensor
_b_ih Tensor
_b_hh

rnn_relu_cell ::
  -- | input
  Tensor ->
  -- | hx
  Tensor ->
  -- | w_ih
  Tensor ->
  -- | w_hh
  Tensor ->
  -- | b_ih
  Tensor ->
  -- | b_hh
  Tensor ->
  Tensor
rnn_relu_cell :: Tensor -> Tensor -> Tensor -> Tensor -> Tensor -> Tensor -> Tensor
rnn_relu_cell Tensor
_input Tensor
_hx Tensor
_w_ih Tensor
_w_hh Tensor
_b_ih Tensor
_b_hh = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> IO (ForeignPtr Tensor)
ATen.rnn_relu_cell_tttttt) Tensor
_input Tensor
_hx Tensor
_w_ih Tensor
_w_hh Tensor
_b_ih Tensor
_b_hh

quantized_lstm_cell ::
  -- | input
  Tensor ->
  -- | hx
  [Tensor] ->
  -- | w_ih
  Tensor ->
  -- | w_hh
  Tensor ->
  -- | b_ih
  Tensor ->
  -- | b_hh
  Tensor ->
  -- | packed_ih
  Tensor ->
  -- | packed_hh
  Tensor ->
  -- | col_offsets_ih
  Tensor ->
  -- | col_offsets_hh
  Tensor ->
  -- | scale_ih
  Float ->
  -- | scale_hh
  Float ->
  -- | zero_point_ih
  Float ->
  -- | zero_point_hh
  Float ->
  (Tensor, Tensor)
quantized_lstm_cell :: Tensor
-> [Tensor]
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Float
-> Float
-> Float
-> Float
-> (Tensor, Tensor)
quantized_lstm_cell Tensor
_input [Tensor]
_hx Tensor
_w_ih Tensor
_w_hh Tensor
_b_ih Tensor
_b_hh Tensor
_packed_ih Tensor
_packed_hh Tensor
_col_offsets_ih Tensor
_col_offsets_hh Float
_scale_ih Float
_scale_hh Float
_zero_point_ih Float
_zero_point_hh = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       x9 cx9 x10 cx10 x11 cx11 x12 cx12 x13 cx13 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable x9 cx9, Castable x10 cx10,
 Castable x11 cx11, Castable x12 cx12, Castable x13 cx13,
 Castable y cy) =>
(ca
 -> cx1
 -> cx2
 -> cx3
 -> cx4
 -> cx5
 -> cx6
 -> cx7
 -> cx8
 -> cx9
 -> cx10
 -> cx11
 -> cx12
 -> cx13
 -> IO cy)
-> a
-> x1
-> x2
-> x3
-> x4
-> x5
-> x6
-> x7
-> x8
-> x9
-> x10
-> x11
-> x12
-> x13
-> IO y
cast14 ForeignPtr Tensor
-> ForeignPtr TensorList
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr Scalar
-> ForeignPtr Scalar
-> ForeignPtr Scalar
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.quantized_lstm_cell_tlttttttttssss) Tensor
_input [Tensor]
_hx Tensor
_w_ih Tensor
_w_hh Tensor
_b_ih Tensor
_b_hh Tensor
_packed_ih Tensor
_packed_hh Tensor
_col_offsets_ih Tensor
_col_offsets_hh Float
_scale_ih Float
_scale_hh Float
_zero_point_ih Float
_zero_point_hh

quantized_gru_cell ::
  -- | input
  Tensor ->
  -- | hx
  Tensor ->
  -- | w_ih
  Tensor ->
  -- | w_hh
  Tensor ->
  -- | b_ih
  Tensor ->
  -- | b_hh
  Tensor ->
  -- | packed_ih
  Tensor ->
  -- | packed_hh
  Tensor ->
  -- | col_offsets_ih
  Tensor ->
  -- | col_offsets_hh
  Tensor ->
  -- | scale_ih
  Float ->
  -- | scale_hh
  Float ->
  -- | zero_point_ih
  Float ->
  -- | zero_point_hh
  Float ->
  Tensor
quantized_gru_cell :: Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Float
-> Float
-> Float
-> Float
-> Tensor
quantized_gru_cell Tensor
_input Tensor
_hx Tensor
_w_ih Tensor
_w_hh Tensor
_b_ih Tensor
_b_hh Tensor
_packed_ih Tensor
_packed_hh Tensor
_col_offsets_ih Tensor
_col_offsets_hh Float
_scale_ih Float
_scale_hh Float
_zero_point_ih Float
_zero_point_hh = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       x9 cx9 x10 cx10 x11 cx11 x12 cx12 x13 cx13 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable x9 cx9, Castable x10 cx10,
 Castable x11 cx11, Castable x12 cx12, Castable x13 cx13,
 Castable y cy) =>
(ca
 -> cx1
 -> cx2
 -> cx3
 -> cx4
 -> cx5
 -> cx6
 -> cx7
 -> cx8
 -> cx9
 -> cx10
 -> cx11
 -> cx12
 -> cx13
 -> IO cy)
-> a
-> x1
-> x2
-> x3
-> x4
-> x5
-> x6
-> x7
-> x8
-> x9
-> x10
-> x11
-> x12
-> x13
-> IO y
cast14 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr Scalar
-> ForeignPtr Scalar
-> ForeignPtr Scalar
-> IO (ForeignPtr Tensor)
ATen.quantized_gru_cell_ttttttttttssss) Tensor
_input Tensor
_hx Tensor
_w_ih Tensor
_w_hh Tensor
_b_ih Tensor
_b_hh Tensor
_packed_ih Tensor
_packed_hh Tensor
_col_offsets_ih Tensor
_col_offsets_hh Float
_scale_ih Float
_scale_hh Float
_zero_point_ih Float
_zero_point_hh

quantized_rnn_relu_cell ::
  -- | input
  Tensor ->
  -- | hx
  Tensor ->
  -- | w_ih
  Tensor ->
  -- | w_hh
  Tensor ->
  -- | b_ih
  Tensor ->
  -- | b_hh
  Tensor ->
  -- | packed_ih
  Tensor ->
  -- | packed_hh
  Tensor ->
  -- | col_offsets_ih
  Tensor ->
  -- | col_offsets_hh
  Tensor ->
  -- | scale_ih
  Float ->
  -- | scale_hh
  Float ->
  -- | zero_point_ih
  Float ->
  -- | zero_point_hh
  Float ->
  Tensor
quantized_rnn_relu_cell :: Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Float
-> Float
-> Float
-> Float
-> Tensor
quantized_rnn_relu_cell Tensor
_input Tensor
_hx Tensor
_w_ih Tensor
_w_hh Tensor
_b_ih Tensor
_b_hh Tensor
_packed_ih Tensor
_packed_hh Tensor
_col_offsets_ih Tensor
_col_offsets_hh Float
_scale_ih Float
_scale_hh Float
_zero_point_ih Float
_zero_point_hh = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       x9 cx9 x10 cx10 x11 cx11 x12 cx12 x13 cx13 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable x9 cx9, Castable x10 cx10,
 Castable x11 cx11, Castable x12 cx12, Castable x13 cx13,
 Castable y cy) =>
(ca
 -> cx1
 -> cx2
 -> cx3
 -> cx4
 -> cx5
 -> cx6
 -> cx7
 -> cx8
 -> cx9
 -> cx10
 -> cx11
 -> cx12
 -> cx13
 -> IO cy)
-> a
-> x1
-> x2
-> x3
-> x4
-> x5
-> x6
-> x7
-> x8
-> x9
-> x10
-> x11
-> x12
-> x13
-> IO y
cast14 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr Scalar
-> ForeignPtr Scalar
-> ForeignPtr Scalar
-> IO (ForeignPtr Tensor)
ATen.quantized_rnn_relu_cell_ttttttttttssss) Tensor
_input Tensor
_hx Tensor
_w_ih Tensor
_w_hh Tensor
_b_ih Tensor
_b_hh Tensor
_packed_ih Tensor
_packed_hh Tensor
_col_offsets_ih Tensor
_col_offsets_hh Float
_scale_ih Float
_scale_hh Float
_zero_point_ih Float
_zero_point_hh

quantized_rnn_tanh_cell ::
  -- | input
  Tensor ->
  -- | hx
  Tensor ->
  -- | w_ih
  Tensor ->
  -- | w_hh
  Tensor ->
  -- | b_ih
  Tensor ->
  -- | b_hh
  Tensor ->
  -- | packed_ih
  Tensor ->
  -- | packed_hh
  Tensor ->
  -- | col_offsets_ih
  Tensor ->
  -- | col_offsets_hh
  Tensor ->
  -- | scale_ih
  Float ->
  -- | scale_hh
  Float ->
  -- | zero_point_ih
  Float ->
  -- | zero_point_hh
  Float ->
  Tensor
quantized_rnn_tanh_cell :: Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Tensor
-> Float
-> Float
-> Float
-> Float
-> Tensor
quantized_rnn_tanh_cell Tensor
_input Tensor
_hx Tensor
_w_ih Tensor
_w_hh Tensor
_b_ih Tensor
_b_hh Tensor
_packed_ih Tensor
_packed_hh Tensor
_col_offsets_ih Tensor
_col_offsets_hh Float
_scale_ih Float
_scale_hh Float
_zero_point_ih Float
_zero_point_hh = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 x8 cx8
       x9 cx9 x10 cx10 x11 cx11 x12 cx12 x13 cx13 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable x8 cx8, Castable x9 cx9, Castable x10 cx10,
 Castable x11 cx11, Castable x12 cx12, Castable x13 cx13,
 Castable y cy) =>
(ca
 -> cx1
 -> cx2
 -> cx3
 -> cx4
 -> cx5
 -> cx6
 -> cx7
 -> cx8
 -> cx9
 -> cx10
 -> cx11
 -> cx12
 -> cx13
 -> IO cy)
-> a
-> x1
-> x2
-> x3
-> x4
-> x5
-> x6
-> x7
-> x8
-> x9
-> x10
-> x11
-> x12
-> x13
-> IO y
cast14 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr Scalar
-> ForeignPtr Scalar
-> ForeignPtr Scalar
-> IO (ForeignPtr Tensor)
ATen.quantized_rnn_tanh_cell_ttttttttttssss) Tensor
_input Tensor
_hx Tensor
_w_ih Tensor
_w_hh Tensor
_b_ih Tensor
_b_hh Tensor
_packed_ih Tensor
_packed_hh Tensor
_col_offsets_ih Tensor
_col_offsets_hh Float
_scale_ih Float
_scale_hh Float
_zero_point_ih Float
_zero_point_hh

maskedFillScalar ::
  -- | self
  Tensor ->
  -- | mask
  Tensor ->
  -- | value
  Float ->
  Tensor
maskedFillScalar :: Tensor -> Tensor -> Float -> Tensor
maskedFillScalar Tensor
_self Tensor
_mask Float
_value = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.masked_fill_tts) Tensor
_self Tensor
_mask Float
_value

maskedFill ::
  -- | self
  Tensor ->
  -- | mask
  Tensor ->
  -- | value
  Tensor ->
  Tensor
maskedFill :: Tensor -> Tensor -> Tensor -> Tensor
maskedFill Tensor
_self Tensor
_mask Tensor
_value = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.masked_fill_ttt) Tensor
_self Tensor
_mask Tensor
_value

masked_scatter ::
  -- | self
  Tensor ->
  -- | mask
  Tensor ->
  -- | source
  Tensor ->
  Tensor
masked_scatter :: Tensor -> Tensor -> Tensor -> Tensor
masked_scatter Tensor
_self Tensor
_mask Tensor
_source = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.masked_scatter_ttt) Tensor
_self Tensor
_mask Tensor
_source

put ::
  -- | self
  Tensor ->
  -- | index
  Tensor ->
  -- | source
  Tensor ->
  -- | accumulate
  Bool ->
  Tensor
put :: Tensor -> Tensor -> Tensor -> Bool -> Tensor
put Tensor
_self Tensor
_index Tensor
_source Bool
_accumulate = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> CBool
-> IO (ForeignPtr Tensor)
ATen.put_tttb) Tensor
_self Tensor
_index Tensor
_source Bool
_accumulate

index_add_tltts ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | index
  Tensor ->
  -- | source
  Tensor ->
  -- | alpha
  Float ->
  Tensor
index_add_tltts :: Tensor -> Int -> Tensor -> Tensor -> Float -> Tensor
index_add_tltts Tensor
_self Int
_dim Tensor
_index Tensor
_source Float
_alpha = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> Int64
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Scalar
-> IO (ForeignPtr Tensor)
ATen.index_add_tltts) Tensor
_self Int
_dim Tensor
_index Tensor
_source Float
_alpha

index_add_tntts ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  -- | index
  Tensor ->
  -- | source
  Tensor ->
  -- | alpha
  Float ->
  Tensor
index_add_tntts :: Tensor -> Dimname -> Tensor -> Tensor -> Float -> Tensor
index_add_tntts Tensor
_self Dimname
_dim Tensor
_index Tensor
_source Float
_alpha = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Dimname
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Scalar
-> IO (ForeignPtr Tensor)
ATen.index_add_tntts) Tensor
_self Dimname
_dim Tensor
_index Tensor
_source Float
_alpha

indexFillScalar ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | index
  Tensor ->
  -- | value
  Float ->
  Tensor
indexFillScalar :: Tensor -> Int -> Tensor -> Float -> Tensor
indexFillScalar Tensor
_self Int
_dim Tensor
_index Float
_value = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64
-> ForeignPtr Tensor
-> ForeignPtr Scalar
-> IO (ForeignPtr Tensor)
ATen.index_fill_tlts) Tensor
_self Int
_dim Tensor
_index Float
_value

indexFill ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | index
  Tensor ->
  -- | value
  Tensor ->
  Tensor
indexFill :: Tensor -> Int -> Tensor -> Tensor -> Tensor
indexFill Tensor
_self Int
_dim Tensor
_index Tensor
_value = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> IO (ForeignPtr Tensor)
ATen.index_fill_tltt) Tensor
_self Int
_dim Tensor
_index Tensor
_value

indexFillScalarWithDimname ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  -- | index
  Tensor ->
  -- | value
  Float ->
  Tensor
indexFillScalarWithDimname :: Tensor -> Dimname -> Tensor -> Float -> Tensor
indexFillScalarWithDimname Tensor
_self Dimname
_dim Tensor
_index Float
_value = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Dimname
-> ForeignPtr Tensor
-> ForeignPtr Scalar
-> IO (ForeignPtr Tensor)
ATen.index_fill_tnts) Tensor
_self Dimname
_dim Tensor
_index Float
_value

indexFillWithDimname ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  -- | index
  Tensor ->
  -- | value
  Tensor ->
  Tensor
indexFillWithDimname :: Tensor -> Dimname -> Tensor -> Tensor -> Tensor
indexFillWithDimname Tensor
_self Dimname
_dim Tensor
_index Tensor
_value = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Dimname
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> IO (ForeignPtr Tensor)
ATen.index_fill_tntt) Tensor
_self Dimname
_dim Tensor
_index Tensor
_value

scatter ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | index
  Tensor ->
  -- | src
  Tensor ->
  Tensor
scatter :: Tensor -> Int -> Tensor -> Tensor -> Tensor
scatter Tensor
_self Int
_dim Tensor
_index Tensor
_src = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> IO (ForeignPtr Tensor)
ATen.scatter_tltt) Tensor
_self Int
_dim Tensor
_index Tensor
_src

scatterScalar ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | index
  Tensor ->
  -- | value
  Float ->
  Tensor
scatterScalar :: Tensor -> Int -> Tensor -> Float -> Tensor
scatterScalar Tensor
_self Int
_dim Tensor
_index Float
_value = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64
-> ForeignPtr Tensor
-> ForeignPtr Scalar
-> IO (ForeignPtr Tensor)
ATen.scatter_tlts) Tensor
_self Int
_dim Tensor
_index Float
_value

scatter_tltts ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | index
  Tensor ->
  -- | src
  Tensor ->
  -- | reduce
  String ->
  Tensor
scatter_tltts :: Tensor -> Int -> Tensor -> Tensor -> String -> Tensor
scatter_tltts Tensor
_self Int
_dim Tensor
_index Tensor
_src String
_reduce = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> Int64
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr StdString
-> IO (ForeignPtr Tensor)
ATen.scatter_tltts) Tensor
_self Int
_dim Tensor
_index Tensor
_src String
_reduce

scatter_tltss ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | index
  Tensor ->
  -- | value
  Float ->
  -- | reduce
  String ->
  Tensor
scatter_tltss :: Tensor -> Int -> Tensor -> Float -> String -> Tensor
scatter_tltss Tensor
_self Int
_dim Tensor
_index Float
_value String
_reduce = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> Int64
-> ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr StdString
-> IO (ForeignPtr Tensor)
ATen.scatter_tltss) Tensor
_self Int
_dim Tensor
_index Float
_value String
_reduce

scatterWithDimname ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  -- | index
  Tensor ->
  -- | src
  Tensor ->
  Tensor
scatterWithDimname :: Tensor -> Dimname -> Tensor -> Tensor -> Tensor
scatterWithDimname Tensor
_self Dimname
_dim Tensor
_index Tensor
_src = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Dimname
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> IO (ForeignPtr Tensor)
ATen.scatter_tntt) Tensor
_self Dimname
_dim Tensor
_index Tensor
_src

scatterScalarWithDimname ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  -- | index
  Tensor ->
  -- | value
  Float ->
  Tensor
scatterScalarWithDimname :: Tensor -> Dimname -> Tensor -> Float -> Tensor
scatterScalarWithDimname Tensor
_self Dimname
_dim Tensor
_index Float
_value = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Dimname
-> ForeignPtr Tensor
-> ForeignPtr Scalar
-> IO (ForeignPtr Tensor)
ATen.scatter_tnts) Tensor
_self Dimname
_dim Tensor
_index Float
_value

scatterAdd ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | index
  Tensor ->
  -- | src
  Tensor ->
  Tensor
scatterAdd :: Tensor -> Int -> Tensor -> Tensor -> Tensor
scatterAdd Tensor
_self Int
_dim Tensor
_index Tensor
_src = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> IO (ForeignPtr Tensor)
ATen.scatter_add_tltt) Tensor
_self Int
_dim Tensor
_index Tensor
_src

scatterAddWithDimname ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  -- | index
  Tensor ->
  -- | src
  Tensor ->
  Tensor
scatterAddWithDimname :: Tensor -> Dimname -> Tensor -> Tensor -> Tensor
scatterAddWithDimname Tensor
_self Dimname
_dim Tensor
_index Tensor
_src = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Dimname
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> IO (ForeignPtr Tensor)
ATen.scatter_add_tntt) Tensor
_self Dimname
_dim Tensor
_index Tensor
_src

scatter_reduce ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | index
  Tensor ->
  -- | reduce
  String ->
  -- | output_size
  Int ->
  Tensor
scatter_reduce :: Tensor -> Int -> Tensor -> String -> Int -> Tensor
scatter_reduce Tensor
_self Int
_dim Tensor
_index String
_reduce Int
_output_size = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> Int64
-> ForeignPtr Tensor
-> ForeignPtr StdString
-> Int64
-> IO (ForeignPtr Tensor)
ATen.scatter_reduce_tltsl) Tensor
_self Int
_dim Tensor
_index String
_reduce Int
_output_size

bitwise_and_ts ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
bitwise_and_ts :: Tensor -> Float -> Tensor
bitwise_and_ts Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.bitwise_and_ts) Tensor
_self Float
_other

bitwise_and_tt ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
bitwise_and_tt :: Tensor -> Tensor -> Tensor
bitwise_and_tt Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.bitwise_and_tt) Tensor
_self Tensor
_other

bitwise_or_ts ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
bitwise_or_ts :: Tensor -> Float -> Tensor
bitwise_or_ts Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.bitwise_or_ts) Tensor
_self Float
_other

bitwise_or_tt ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
bitwise_or_tt :: Tensor -> Tensor -> Tensor
bitwise_or_tt Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.bitwise_or_tt) Tensor
_self Tensor
_other

bitwiseXorScalar ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
bitwiseXorScalar :: Tensor -> Float -> Tensor
bitwiseXorScalar Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.bitwise_xor_ts) Tensor
_self Float
_other

bitwiseXor ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
bitwiseXor :: Tensor -> Tensor -> Tensor
bitwiseXor Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.bitwise_xor_tt) Tensor
_self Tensor
_other

bitwise_left_shift_tt ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
bitwise_left_shift_tt :: Tensor -> Tensor -> Tensor
bitwise_left_shift_tt Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.bitwise_left_shift_tt) Tensor
_self Tensor
_other

bitwise_left_shift_ts ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
bitwise_left_shift_ts :: Tensor -> Float -> Tensor
bitwise_left_shift_ts Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.bitwise_left_shift_ts) Tensor
_self Float
_other

bitwise_left_shift_st ::
  -- | self
  Float ->
  -- | other
  Tensor ->
  Tensor
bitwise_left_shift_st :: Float -> Tensor -> Tensor
bitwise_left_shift_st Float
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Scalar -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.bitwise_left_shift_st) Float
_self Tensor
_other

bitwise_right_shift_tt ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
bitwise_right_shift_tt :: Tensor -> Tensor -> Tensor
bitwise_right_shift_tt Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.bitwise_right_shift_tt) Tensor
_self Tensor
_other

bitwise_right_shift_ts ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
bitwise_right_shift_ts :: Tensor -> Float -> Tensor
bitwise_right_shift_ts Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.bitwise_right_shift_ts) Tensor
_self Float
_other

bitwise_right_shift_st ::
  -- | self
  Float ->
  -- | other
  Tensor ->
  Tensor
bitwise_right_shift_st :: Float -> Tensor -> Tensor
bitwise_right_shift_st Float
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Scalar -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.bitwise_right_shift_st) Float
_self Tensor
_other

addbmm ::
  -- | self
  Tensor ->
  -- | batch1
  Tensor ->
  -- | batch2
  Tensor ->
  -- | beta
  Float ->
  -- | alpha
  Float ->
  Tensor
addbmm :: Tensor -> Tensor -> Tensor -> Float -> Float -> Tensor
addbmm Tensor
_self Tensor
_batch1 Tensor
_batch2 Float
_beta Float
_alpha = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr Scalar
-> IO (ForeignPtr Tensor)
ATen.addbmm_tttss) Tensor
_self Tensor
_batch1 Tensor
_batch2 Float
_beta Float
_alpha

diag ::
  -- | self
  Tensor ->
  -- | diagonal
  Int ->
  Tensor
diag :: Tensor -> Int -> Tensor
diag Tensor
_self Int
_diagonal = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.diag_tl) Tensor
_self Int
_diagonal

cross ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  -- | dim
  Int ->
  Tensor
cross :: Tensor -> Tensor -> Int -> Tensor
cross Tensor
_self Tensor
_other Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.cross_ttl) Tensor
_self Tensor
_other Int
_dim

triu ::
  -- | self
  Tensor ->
  -- | diagonal
  Int ->
  Tensor
triu :: Tensor -> Int -> Tensor
triu Tensor
_self Int
_diagonal = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.triu_tl) Tensor
_self Int
_diagonal

tril ::
  -- | self
  Tensor ->
  -- | diagonal
  Int ->
  Tensor
tril :: Tensor -> Int -> Tensor
tril Tensor
_self Int
_diagonal = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.tril_tl) Tensor
_self Int
_diagonal

trace ::
  -- | self
  Tensor ->
  Tensor
trace :: Tensor -> Tensor
trace Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.trace_t) Tensor
_self

neScalar ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
neScalar :: Tensor -> Float -> Tensor
neScalar Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.ne_ts) Tensor
_self Float
_other

ne ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
ne :: Tensor -> Tensor -> Tensor
ne Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.ne_tt) Tensor
_self Tensor
_other

not_equal_ts ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
not_equal_ts :: Tensor -> Float -> Tensor
not_equal_ts Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.not_equal_ts) Tensor
_self Float
_other

not_equal_tt ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
not_equal_tt :: Tensor -> Tensor -> Tensor
not_equal_tt Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.not_equal_tt) Tensor
_self Tensor
_other

eqScalar ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
eqScalar :: Tensor -> Float -> Tensor
eqScalar Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.eq_ts) Tensor
_self Float
_other

eq ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
eq :: Tensor -> Tensor -> Tensor
eq Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.eq_tt) Tensor
_self Tensor
_other

geScalar ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
geScalar :: Tensor -> Float -> Tensor
geScalar Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.ge_ts) Tensor
_self Float
_other

ge ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
ge :: Tensor -> Tensor -> Tensor
ge Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.ge_tt) Tensor
_self Tensor
_other

greater_equal_ts ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
greater_equal_ts :: Tensor -> Float -> Tensor
greater_equal_ts Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.greater_equal_ts) Tensor
_self Float
_other

greater_equal_tt ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
greater_equal_tt :: Tensor -> Tensor -> Tensor
greater_equal_tt Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.greater_equal_tt) Tensor
_self Tensor
_other

leScalar ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
leScalar :: Tensor -> Float -> Tensor
leScalar Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.le_ts) Tensor
_self Float
_other

le ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
le :: Tensor -> Tensor -> Tensor
le Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.le_tt) Tensor
_self Tensor
_other

less_equal_ts ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
less_equal_ts :: Tensor -> Float -> Tensor
less_equal_ts Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.less_equal_ts) Tensor
_self Float
_other

less_equal_tt ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
less_equal_tt :: Tensor -> Tensor -> Tensor
less_equal_tt Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.less_equal_tt) Tensor
_self Tensor
_other

gtScalar ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
gtScalar :: Tensor -> Float -> Tensor
gtScalar Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.gt_ts) Tensor
_self Float
_other

gt ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
gt :: Tensor -> Tensor -> Tensor
gt Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.gt_tt) Tensor
_self Tensor
_other

greater_ts ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
greater_ts :: Tensor -> Float -> Tensor
greater_ts Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.greater_ts) Tensor
_self Float
_other

greater_tt ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
greater_tt :: Tensor -> Tensor -> Tensor
greater_tt Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.greater_tt) Tensor
_self Tensor
_other

ltScalar ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
ltScalar :: Tensor -> Float -> Tensor
ltScalar Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.lt_ts) Tensor
_self Float
_other

lt ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
lt :: Tensor -> Tensor -> Tensor
lt Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.lt_tt) Tensor
_self Tensor
_other

less_ts ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
less_ts :: Tensor -> Float -> Tensor
less_ts Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.less_ts) Tensor
_self Float
_other

less_tt ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
less_tt :: Tensor -> Tensor -> Tensor
less_tt Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.less_tt) Tensor
_self Tensor
_other

take ::
  -- | self
  Tensor ->
  -- | index
  Tensor ->
  Tensor
take :: Tensor -> Tensor -> Tensor
take Tensor
_self Tensor
_index = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.take_tt) Tensor
_self Tensor
_index

take_along_dim ::
  -- | self
  Tensor ->
  -- | indices
  Tensor ->
  -- | dim
  Int ->
  Tensor
take_along_dim :: Tensor -> Tensor -> Int -> Tensor
take_along_dim Tensor
_self Tensor
_indices Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.take_along_dim_ttl) Tensor
_self Tensor
_indices Int
_dim

indexSelect ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | index
  Tensor ->
  Tensor
indexSelect :: Tensor -> Int -> Tensor -> Tensor
indexSelect Tensor
_self Int
_dim Tensor
_index = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> Int64 -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.index_select_tlt) Tensor
_self Int
_dim Tensor
_index

indexSelectWithDimname ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  -- | index
  Tensor ->
  Tensor
indexSelectWithDimname :: Tensor -> Dimname -> Tensor -> Tensor
indexSelectWithDimname Tensor
_self Dimname
_dim Tensor
_index = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Dimname
-> ForeignPtr Tensor
-> IO (ForeignPtr Tensor)
ATen.index_select_tnt) Tensor
_self Dimname
_dim Tensor
_index

masked_select ::
  -- | self
  Tensor ->
  -- | mask
  Tensor ->
  Tensor
masked_select :: Tensor -> Tensor -> Tensor
masked_select Tensor
_self Tensor
_mask = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.masked_select_tt) Tensor
_self Tensor
_mask

nonzero ::
  -- | self
  Tensor ->
  Tensor
nonzero :: Tensor -> Tensor
nonzero Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.nonzero_t) Tensor
_self

nonzero_numpy ::
  -- | self
  Tensor ->
  [Tensor]
nonzero_numpy :: Tensor -> [Tensor]
nonzero_numpy Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr TensorList)
ATen.nonzero_numpy_t) Tensor
_self

argwhere ::
  -- | self
  Tensor ->
  Tensor
argwhere :: Tensor -> Tensor
argwhere Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.argwhere_t) Tensor
_self

gather ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | index
  Tensor ->
  -- | sparse_grad
  Bool ->
  Tensor
gather :: Tensor -> Int -> Tensor -> Bool -> Tensor
gather Tensor
_self Int
_dim Tensor
_index Bool
_sparse_grad = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64 -> ForeignPtr Tensor -> CBool -> IO (ForeignPtr Tensor)
ATen.gather_tltb) Tensor
_self Int
_dim Tensor
_index Bool
_sparse_grad

gatherWithDimname ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  -- | index
  Tensor ->
  -- | sparse_grad
  Bool ->
  Tensor
gatherWithDimname :: Tensor -> Dimname -> Tensor -> Bool -> Tensor
gatherWithDimname Tensor
_self Dimname
_dim Tensor
_index Bool
_sparse_grad = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Dimname
-> ForeignPtr Tensor
-> CBool
-> IO (ForeignPtr Tensor)
ATen.gather_tntb) Tensor
_self Dimname
_dim Tensor
_index Bool
_sparse_grad

addcmul ::
  -- | self
  Tensor ->
  -- | tensor1
  Tensor ->
  -- | tensor2
  Tensor ->
  -- | value
  Float ->
  Tensor
addcmul :: Tensor -> Tensor -> Tensor -> Float -> Tensor
addcmul Tensor
_self Tensor
_tensor1 Tensor
_tensor2 Float
_value = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Scalar
-> IO (ForeignPtr Tensor)
ATen.addcmul_ttts) Tensor
_self Tensor
_tensor1 Tensor
_tensor2 Float
_value

addcdiv ::
  -- | self
  Tensor ->
  -- | tensor1
  Tensor ->
  -- | tensor2
  Tensor ->
  -- | value
  Float ->
  Tensor
addcdiv :: Tensor -> Tensor -> Tensor -> Float -> Tensor
addcdiv Tensor
_self Tensor
_tensor1 Tensor
_tensor2 Float
_value = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Scalar
-> IO (ForeignPtr Tensor)
ATen.addcdiv_ttts) Tensor
_self Tensor
_tensor1 Tensor
_tensor2 Float
_value

cross_entropy_loss ::
  -- | self
  Tensor ->
  -- | target
  Tensor ->
  -- | weight
  Tensor ->
  -- | reduction
  Int ->
  -- | ignore_index
  Int ->
  -- | label_smoothing
  Double ->
  Tensor
cross_entropy_loss :: Tensor -> Tensor -> Tensor -> Int -> Int -> Double -> Tensor
cross_entropy_loss Tensor
_self Tensor
_target Tensor
_weight Int
_reduction Int
_ignore_index Double
_label_smoothing = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> Int64
-> Int64
-> CDouble
-> IO (ForeignPtr Tensor)
ATen.cross_entropy_loss_tttlld) Tensor
_self Tensor
_target Tensor
_weight Int
_reduction Int
_ignore_index Double
_label_smoothing

lstsq ::
  -- | self
  Tensor ->
  -- | A
  Tensor ->
  (Tensor, Tensor)
lstsq :: Tensor -> Tensor -> (Tensor, Tensor)
lstsq Tensor
_self Tensor
_A = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor
-> ForeignPtr Tensor
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.lstsq_tt) Tensor
_self Tensor
_A

triangular_solve ::
  -- | self
  Tensor ->
  -- | A
  Tensor ->
  -- | upper
  Bool ->
  -- | transpose
  Bool ->
  -- | unitriangular
  Bool ->
  (Tensor, Tensor)
triangular_solve :: Tensor -> Tensor -> Bool -> Bool -> Bool -> (Tensor, Tensor)
triangular_solve Tensor
_self Tensor
_A Bool
_upper Bool
_transpose Bool
_unitriangular = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> CBool
-> CBool
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.triangular_solve_ttbbb) Tensor
_self Tensor
_A Bool
_upper Bool
_transpose Bool
_unitriangular

linalg_solve_triangular ::
  -- | self
  Tensor ->
  -- | B
  Tensor ->
  -- | upper
  Bool ->
  -- | left
  Bool ->
  -- | unitriangular
  Bool ->
  Tensor
linalg_solve_triangular :: Tensor -> Tensor -> Bool -> Bool -> Bool -> Tensor
linalg_solve_triangular Tensor
_self Tensor
_B Bool
_upper Bool
_left Bool
_unitriangular = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> CBool
-> CBool
-> CBool
-> IO (ForeignPtr Tensor)
ATen.linalg_solve_triangular_ttbbb) Tensor
_self Tensor
_B Bool
_upper Bool
_left Bool
_unitriangular

symeig ::
  -- | self
  Tensor ->
  -- | eigenvectors
  Bool ->
  -- | upper
  Bool ->
  (Tensor, Tensor)
symeig :: Tensor -> Bool -> Bool -> (Tensor, Tensor)
symeig Tensor
_self Bool
_eigenvectors Bool
_upper = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> CBool -> CBool -> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.symeig_tbb) Tensor
_self Bool
_eigenvectors Bool
_upper

eig ::
  -- | self
  Tensor ->
  -- | eigenvectors
  Bool ->
  (Tensor, Tensor)
eig :: Tensor -> Bool -> (Tensor, Tensor)
eig Tensor
_self Bool
_eigenvectors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor
-> CBool -> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.eig_tb) Tensor
_self Bool
_eigenvectors

svd ::
  -- | self
  Tensor ->
  -- | some
  Bool ->
  -- | compute_uv
  Bool ->
  (Tensor, Tensor, Tensor)
svd :: Tensor -> Bool -> Bool -> (Tensor, Tensor, Tensor)
svd Tensor
_self Bool
_some Bool
_compute_uv = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> CBool
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor, Tensor)))
ATen.svd_tbb) Tensor
_self Bool
_some Bool
_compute_uv

swapaxes ::
  -- | self
  Tensor ->
  -- | axis0
  Int ->
  -- | axis1
  Int ->
  Tensor
swapaxes :: Tensor -> Int -> Int -> Tensor
swapaxes Tensor
_self Int
_axis0 Int
_axis1 = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> Int64 -> Int64 -> IO (ForeignPtr Tensor)
ATen.swapaxes_tll) Tensor
_self Int
_axis0 Int
_axis1

swapdims ::
  -- | self
  Tensor ->
  -- | dim0
  Int ->
  -- | dim1
  Int ->
  Tensor
swapdims :: Tensor -> Int -> Int -> Tensor
swapdims Tensor
_self Int
_dim0 Int
_dim1 = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> Int64 -> Int64 -> IO (ForeignPtr Tensor)
ATen.swapdims_tll) Tensor
_self Int
_dim0 Int
_dim1

cholesky ::
  -- | self
  Tensor ->
  -- | upper
  Bool ->
  Tensor
cholesky :: Tensor -> Bool -> Tensor
cholesky Tensor
_self Bool
_upper = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> CBool -> IO (ForeignPtr Tensor)
ATen.cholesky_tb) Tensor
_self Bool
_upper

cholesky_solve ::
  -- | self
  Tensor ->
  -- | input2
  Tensor ->
  -- | upper
  Bool ->
  Tensor
cholesky_solve :: Tensor -> Tensor -> Bool -> Tensor
cholesky_solve Tensor
_self Tensor
_input2 Bool
_upper = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> CBool -> IO (ForeignPtr Tensor)
ATen.cholesky_solve_ttb) Tensor
_self Tensor
_input2 Bool
_upper

solve ::
  -- | self
  Tensor ->
  -- | A
  Tensor ->
  (Tensor, Tensor)
solve :: Tensor -> Tensor -> (Tensor, Tensor)
solve Tensor
_self Tensor
_A = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor
-> ForeignPtr Tensor
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.solve_tt) Tensor
_self Tensor
_A

cholesky_inverse ::
  -- | self
  Tensor ->
  -- | upper
  Bool ->
  Tensor
cholesky_inverse :: Tensor -> Bool -> Tensor
cholesky_inverse Tensor
_self Bool
_upper = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> CBool -> IO (ForeignPtr Tensor)
ATen.cholesky_inverse_tb) Tensor
_self Bool
_upper

qr ::
  -- | self
  Tensor ->
  -- | some
  Bool ->
  (Tensor, Tensor)
qr :: Tensor -> Bool -> (Tensor, Tensor)
qr Tensor
_self Bool
_some = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor
-> CBool -> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.qr_tb) Tensor
_self Bool
_some

geqrf ::
  -- | self
  Tensor ->
  (Tensor, Tensor)
geqrf :: Tensor -> (Tensor, Tensor)
geqrf Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.geqrf_t) Tensor
_self

orgqr ::
  -- | self
  Tensor ->
  -- | input2
  Tensor ->
  Tensor
orgqr :: Tensor -> Tensor -> Tensor
orgqr Tensor
_self Tensor
_input2 = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.orgqr_tt) Tensor
_self Tensor
_input2

ormqr ::
  -- | self
  Tensor ->
  -- | input2
  Tensor ->
  -- | input3
  Tensor ->
  -- | left
  Bool ->
  -- | transpose
  Bool ->
  Tensor
ormqr :: Tensor -> Tensor -> Tensor -> Bool -> Bool -> Tensor
ormqr Tensor
_self Tensor
_input2 Tensor
_input3 Bool
_left Bool
_transpose = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> CBool
-> CBool
-> IO (ForeignPtr Tensor)
ATen.ormqr_tttbb) Tensor
_self Tensor
_input2 Tensor
_input3 Bool
_left Bool
_transpose

lu_solve ::
  -- | self
  Tensor ->
  -- | LU_data
  Tensor ->
  -- | LU_pivots
  Tensor ->
  Tensor
lu_solve :: Tensor -> Tensor -> Tensor -> Tensor
lu_solve Tensor
_self Tensor
_LU_data Tensor
_LU_pivots = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.lu_solve_ttt) Tensor
_self Tensor
_LU_data Tensor
_LU_pivots

lu_unpack ::
  -- | LU_data
  Tensor ->
  -- | LU_pivots
  Tensor ->
  -- | unpack_data
  Bool ->
  -- | unpack_pivots
  Bool ->
  (Tensor, Tensor, Tensor)
lu_unpack :: Tensor -> Tensor -> Bool -> Bool -> (Tensor, Tensor, Tensor)
lu_unpack Tensor
_LU_data Tensor
_LU_pivots Bool
_unpack_data Bool
_unpack_pivots = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Tensor
-> CBool
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor, Tensor)))
ATen.lu_unpack_ttbb) Tensor
_LU_data Tensor
_LU_pivots Bool
_unpack_data Bool
_unpack_pivots

lgamma ::
  -- | self
  Tensor ->
  Tensor
lgamma :: Tensor -> Tensor
lgamma Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.lgamma_t) Tensor
_self

digamma ::
  -- | self
  Tensor ->
  Tensor
digamma :: Tensor -> Tensor
digamma Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.digamma_t) Tensor
_self

polygamma ::
  -- | n
  Int ->
  -- | self
  Tensor ->
  Tensor
polygamma :: Int -> Tensor -> Tensor
polygamma Int
_n Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 Int64 -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.polygamma_lt) Int
_n Tensor
_self

erfinv ::
  -- | self
  Tensor ->
  Tensor
erfinv :: Tensor -> Tensor
erfinv Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.erfinv_t) Tensor
_self

i0 ::
  -- | self
  Tensor ->
  Tensor
i0 :: Tensor -> Tensor
i0 Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.i0_t) Tensor
_self

sign ::
  -- | self
  Tensor ->
  Tensor
sign :: Tensor -> Tensor
sign Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.sign_t) Tensor
_self

signbit ::
  -- | self
  Tensor ->
  Tensor
signbit :: Tensor -> Tensor
signbit Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.signbit_t) Tensor
_self

dist ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  -- | p
  Float ->
  Tensor
dist :: Tensor -> Tensor -> Float -> Tensor
dist Tensor
_self Tensor
_other Float
_p = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.dist_tts) Tensor
_self Tensor
_other Float
_p

atan2 ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
atan2 :: Tensor -> Tensor -> Tensor
atan2 Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.atan2_tt) Tensor
_self Tensor
_other

arctan2 ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
arctan2 :: Tensor -> Tensor -> Tensor
arctan2 Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.arctan2_tt) Tensor
_self Tensor
_other

lerpScalar ::
  -- | self
  Tensor ->
  -- | end
  Tensor ->
  -- | weight
  Float ->
  Tensor
lerpScalar :: Tensor -> Tensor -> Float -> Tensor
lerpScalar Tensor
_self Tensor
_end Float
_weight = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.lerp_tts) Tensor
_self Tensor
_end Float
_weight

lerp ::
  -- | self
  Tensor ->
  -- | end
  Tensor ->
  -- | weight
  Tensor ->
  Tensor
lerp :: Tensor -> Tensor -> Tensor -> Tensor
lerp Tensor
_self Tensor
_end Tensor
_weight = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.lerp_ttt) Tensor
_self Tensor
_end Tensor
_weight

histc ::
  -- | self
  Tensor ->
  -- | bins
  Int ->
  -- | min
  Float ->
  -- | max
  Float ->
  Tensor
histc :: Tensor -> Int -> Float -> Float -> Tensor
histc Tensor
_self Int
_bins Float
_min Float
_max = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64
-> ForeignPtr Scalar
-> ForeignPtr Scalar
-> IO (ForeignPtr Tensor)
ATen.histc_tlss) Tensor
_self Int
_bins Float
_min Float
_max

histogram_tttb ::
  -- | self
  Tensor ->
  -- | bins
  Tensor ->
  -- | weight
  Tensor ->
  -- | density
  Bool ->
  (Tensor, Tensor)
histogram_tttb :: Tensor -> Tensor -> Tensor -> Bool -> (Tensor, Tensor)
histogram_tttb Tensor
_self Tensor
_bins Tensor
_weight Bool
_density = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.histogram_tttb) Tensor
_self Tensor
_bins Tensor
_weight Bool
_density

histogram_tlatb ::
  -- | self
  Tensor ->
  -- | bins
  Int ->
  -- | range
  ([Double]) ->
  -- | weight
  Tensor ->
  -- | density
  Bool ->
  (Tensor, Tensor)
histogram_tlatb :: Tensor -> Int -> [Double] -> Tensor -> Bool -> (Tensor, Tensor)
histogram_tlatb Tensor
_self Int
_bins [Double]
_range Tensor
_weight Bool
_density = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> Int64
-> ForeignPtr (StdVector CDouble)
-> ForeignPtr Tensor
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.histogram_tlatb) Tensor
_self Int
_bins [Double]
_range Tensor
_weight Bool
_density

fmodScalar ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
fmodScalar :: Tensor -> Float -> Tensor
fmodScalar Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.fmod_ts) Tensor
_self Float
_other

fmod ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
fmod :: Tensor -> Tensor -> Tensor
fmod Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.fmod_tt) Tensor
_self Tensor
_other

hypot ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
hypot :: Tensor -> Tensor -> Tensor
hypot Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.hypot_tt) Tensor
_self Tensor
_other

igamma ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
igamma :: Tensor -> Tensor -> Tensor
igamma Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.igamma_tt) Tensor
_self Tensor
_other

igammac ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
igammac :: Tensor -> Tensor -> Tensor
igammac Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.igammac_tt) Tensor
_self Tensor
_other

nextafter ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
nextafter :: Tensor -> Tensor -> Tensor
nextafter Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.nextafter_tt) Tensor
_self Tensor
_other

remainderScalar ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
remainderScalar :: Tensor -> Float -> Tensor
remainderScalar Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.remainder_ts) Tensor
_self Float
_other

remainder ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
remainder :: Tensor -> Tensor -> Tensor
remainder Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.remainder_tt) Tensor
_self Tensor
_other

remainder_st ::
  -- | self
  Float ->
  -- | other
  Tensor ->
  Tensor
remainder_st :: Float -> Tensor -> Tensor
remainder_st Float
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Scalar -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.remainder_st) Float
_self Tensor
_other

minAll ::
  -- | self
  Tensor ->
  Tensor
minAll :: Tensor -> Tensor
minAll Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.min_t) Tensor
_self

fmin ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
fmin :: Tensor -> Tensor -> Tensor
fmin Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.fmin_tt) Tensor
_self Tensor
_other

maxAll ::
  -- | self
  Tensor ->
  Tensor
maxAll :: Tensor -> Tensor
maxAll Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.max_t) Tensor
_self

fmax ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
fmax :: Tensor -> Tensor -> Tensor
fmax Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.fmax_tt) Tensor
_self Tensor
_other

maximum ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
maximum :: Tensor -> Tensor -> Tensor
maximum Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.maximum_tt) Tensor
_self Tensor
_other

max ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
max :: Tensor -> Tensor -> Tensor
max Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.max_tt) Tensor
_self Tensor
_other

minimum ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
minimum :: Tensor -> Tensor -> Tensor
minimum Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.minimum_tt) Tensor
_self Tensor
_other

min ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
min :: Tensor -> Tensor -> Tensor
min Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.min_tt) Tensor
_self Tensor
_other

quantile_ttlbs ::
  -- | self
  Tensor ->
  -- | q
  Tensor ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  -- | interpolation
  String ->
  Tensor
quantile_ttlbs :: Tensor -> Tensor -> Int -> Bool -> String -> Tensor
quantile_ttlbs Tensor
_self Tensor
_q Int
_dim Bool
_keepdim String
_interpolation = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> Int64
-> CBool
-> ForeignPtr StdString
-> IO (ForeignPtr Tensor)
ATen.quantile_ttlbs) Tensor
_self Tensor
_q Int
_dim Bool
_keepdim String
_interpolation

quantile_tdlbs ::
  -- | self
  Tensor ->
  -- | q
  Double ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  -- | interpolation
  String ->
  Tensor
quantile_tdlbs :: Tensor -> Double -> Int -> Bool -> String -> Tensor
quantile_tdlbs Tensor
_self Double
_q Int
_dim Bool
_keepdim String
_interpolation = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> CDouble
-> Int64
-> CBool
-> ForeignPtr StdString
-> IO (ForeignPtr Tensor)
ATen.quantile_tdlbs) Tensor
_self Double
_q Int
_dim Bool
_keepdim String
_interpolation

nanquantile_ttlbs ::
  -- | self
  Tensor ->
  -- | q
  Tensor ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  -- | interpolation
  String ->
  Tensor
nanquantile_ttlbs :: Tensor -> Tensor -> Int -> Bool -> String -> Tensor
nanquantile_ttlbs Tensor
_self Tensor
_q Int
_dim Bool
_keepdim String
_interpolation = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> Int64
-> CBool
-> ForeignPtr StdString
-> IO (ForeignPtr Tensor)
ATen.nanquantile_ttlbs) Tensor
_self Tensor
_q Int
_dim Bool
_keepdim String
_interpolation

nanquantile_tdlbs ::
  -- | self
  Tensor ->
  -- | q
  Double ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  -- | interpolation
  String ->
  Tensor
nanquantile_tdlbs :: Tensor -> Double -> Int -> Bool -> String -> Tensor
nanquantile_tdlbs Tensor
_self Double
_q Int
_dim Bool
_keepdim String
_interpolation = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> CDouble
-> Int64
-> CBool
-> ForeignPtr StdString
-> IO (ForeignPtr Tensor)
ATen.nanquantile_tdlbs) Tensor
_self Double
_q Int
_dim Bool
_keepdim String
_interpolation

sort ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | descending
  Bool ->
  (Tensor, Tensor)
sort :: Tensor -> Int -> Bool -> (Tensor, Tensor)
sort Tensor
_self Int
_dim Bool
_descending = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> Int64 -> CBool -> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.sort_tlb) Tensor
_self Int
_dim Bool
_descending

sort_tblb ::
  -- | self
  Tensor ->
  -- | stable
  Bool ->
  -- | dim
  Int ->
  -- | descending
  Bool ->
  (Tensor, Tensor)
sort_tblb :: Tensor -> Bool -> Int -> Bool -> (Tensor, Tensor)
sort_tblb Tensor
_self Bool
_stable Int
_dim Bool
_descending = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> CBool
-> Int64
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.sort_tblb) Tensor
_self Bool
_stable Int
_dim Bool
_descending

sortWithDimname ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  -- | descending
  Bool ->
  (Tensor, Tensor)
sortWithDimname :: Tensor -> Dimname -> Bool -> (Tensor, Tensor)
sortWithDimname Tensor
_self Dimname
_dim Bool
_descending = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Dimname
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.sort_tnb) Tensor
_self Dimname
_dim Bool
_descending

sort_tbnb ::
  -- | self
  Tensor ->
  -- | stable
  Bool ->
  -- | dim
  Dimname ->
  -- | descending
  Bool ->
  (Tensor, Tensor)
sort_tbnb :: Tensor -> Bool -> Dimname -> Bool -> (Tensor, Tensor)
sort_tbnb Tensor
_self Bool
_stable Dimname
_dim Bool
_descending = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> CBool
-> ForeignPtr Dimname
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.sort_tbnb) Tensor
_self Bool
_stable Dimname
_dim Bool
_descending

msort ::
  -- | self
  Tensor ->
  Tensor
msort :: Tensor -> Tensor
msort Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.msort_t) Tensor
_self

argsort ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | descending
  Bool ->
  Tensor
argsort :: Tensor -> Int -> Bool -> Tensor
argsort Tensor
_self Int
_dim Bool
_descending = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> Int64 -> CBool -> IO (ForeignPtr Tensor)
ATen.argsort_tlb) Tensor
_self Int
_dim Bool
_descending

argsortWithDimname ::
  -- | self
  Tensor ->
  -- | dim
  Dimname ->
  -- | descending
  Bool ->
  Tensor
argsortWithDimname :: Tensor -> Dimname -> Bool -> Tensor
argsortWithDimname Tensor
_self Dimname
_dim Bool
_descending = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Dimname -> CBool -> IO (ForeignPtr Tensor)
ATen.argsort_tnb) Tensor
_self Dimname
_dim Bool
_descending

topk ::
  -- | self
  Tensor ->
  -- | k
  Int ->
  -- | dim
  Int ->
  -- | largest
  Bool ->
  -- | sorted
  Bool ->
  (Tensor, Tensor)
topk :: Tensor -> Int -> Int -> Bool -> Bool -> (Tensor, Tensor)
topk Tensor
_self Int
_k Int
_dim Bool
_largest Bool
_sorted = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> Int64
-> Int64
-> CBool
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.topk_tllbb) Tensor
_self Int
_k Int
_dim Bool
_largest Bool
_sorted

all ::
  -- | self
  Tensor ->
  Tensor
all :: Tensor -> Tensor
all Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.all_t) Tensor
_self

any ::
  -- | self
  Tensor ->
  Tensor
any :: Tensor -> Tensor
any Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.any_t) Tensor
_self

renorm ::
  -- | self
  Tensor ->
  -- | p
  Float ->
  -- | dim
  Int ->
  -- | maxnorm
  Float ->
  Tensor
renorm :: Tensor -> Float -> Int -> Float -> Tensor
renorm Tensor
_self Float
_p Int
_dim Float
_maxnorm = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Scalar
-> Int64
-> ForeignPtr Scalar
-> IO (ForeignPtr Tensor)
ATen.renorm_tsls) Tensor
_self Float
_p Int
_dim Float
_maxnorm

equal ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Bool
equal :: Tensor -> Tensor -> Bool
equal Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO CBool
ATen.equal_tt) Tensor
_self Tensor
_other

pow ::
  -- | self
  Tensor ->
  -- | exponent
  Tensor ->
  Tensor
pow :: Tensor -> Tensor -> Tensor
pow Tensor
_self Tensor
_exponent = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.pow_tt) Tensor
_self Tensor
_exponent

powScalar' ::
  -- | self
  Float ->
  -- | exponent
  Tensor ->
  Tensor
powScalar' :: Float -> Tensor -> Tensor
powScalar' Float
_self Tensor
_exponent = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Scalar -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.pow_st) Float
_self Tensor
_exponent

powScalar ::
  -- | self
  Tensor ->
  -- | exponent
  Float ->
  Tensor
powScalar :: Tensor -> Float -> Tensor
powScalar Tensor
_self Float
_exponent = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.pow_ts) Tensor
_self Float
_exponent

float_power_tt ::
  -- | self
  Tensor ->
  -- | exponent
  Tensor ->
  Tensor
float_power_tt :: Tensor -> Tensor -> Tensor
float_power_tt Tensor
_self Tensor
_exponent = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.float_power_tt) Tensor
_self Tensor
_exponent

float_power_st ::
  -- | self
  Float ->
  -- | exponent
  Tensor ->
  Tensor
float_power_st :: Float -> Tensor -> Tensor
float_power_st Float
_self Tensor
_exponent = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Scalar -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.float_power_st) Float
_self Tensor
_exponent

float_power_ts ::
  -- | self
  Tensor ->
  -- | exponent
  Float ->
  Tensor
float_power_ts :: Tensor -> Float -> Tensor
float_power_ts Tensor
_self Float
_exponent = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.float_power_ts) Tensor
_self Float
_exponent

alias ::
  -- | self
  Tensor ->
  Tensor
alias :: Tensor -> Tensor
alias Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.alias_t) Tensor
_self

bucketize_ttbb ::
  -- | self
  Tensor ->
  -- | boundaries
  Tensor ->
  -- | out_int32
  Bool ->
  -- | right
  Bool ->
  Tensor
bucketize_ttbb :: Tensor -> Tensor -> Bool -> Bool -> Tensor
bucketize_ttbb Tensor
_self Tensor
_boundaries Bool
_out_int32 Bool
_right = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Tensor -> CBool -> CBool -> IO (ForeignPtr Tensor)
ATen.bucketize_ttbb) Tensor
_self Tensor
_boundaries Bool
_out_int32 Bool
_right

bucketize_stbb ::
  -- | self
  Float ->
  -- | boundaries
  Tensor ->
  -- | out_int32
  Bool ->
  -- | right
  Bool ->
  Tensor
bucketize_stbb :: Float -> Tensor -> Bool -> Bool -> Tensor
bucketize_stbb Float
_self Tensor
_boundaries Bool
_out_int32 Bool
_right = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Scalar
-> ForeignPtr Tensor -> CBool -> CBool -> IO (ForeignPtr Tensor)
ATen.bucketize_stbb) Float
_self Tensor
_boundaries Bool
_out_int32 Bool
_right

searchsorted_ttbbst ::
  -- | sorted_sequence
  Tensor ->
  -- | self
  Tensor ->
  -- | out_int32
  Bool ->
  -- | right
  Bool ->
  -- | side
  String ->
  -- | sorter
  Tensor ->
  Tensor
searchsorted_ttbbst :: Tensor -> Tensor -> Bool -> Bool -> String -> Tensor -> Tensor
searchsorted_ttbbst Tensor
_sorted_sequence Tensor
_self Bool
_out_int32 Bool
_right String
_side Tensor
_sorter = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr Tensor
-> CBool
-> CBool
-> ForeignPtr StdString
-> ForeignPtr Tensor
-> IO (ForeignPtr Tensor)
ATen.searchsorted_ttbbst) Tensor
_sorted_sequence Tensor
_self Bool
_out_int32 Bool
_right String
_side Tensor
_sorter

searchsorted_tsbbst ::
  -- | sorted_sequence
  Tensor ->
  -- | self
  Float ->
  -- | out_int32
  Bool ->
  -- | right
  Bool ->
  -- | side
  String ->
  -- | sorter
  Tensor ->
  Tensor
searchsorted_tsbbst :: Tensor -> Float -> Bool -> Bool -> String -> Tensor -> Tensor
searchsorted_tsbbst Tensor
_sorted_sequence Float
_self Bool
_out_int32 Bool
_right String
_side Tensor
_sorter = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr Scalar
-> CBool
-> CBool
-> ForeignPtr StdString
-> ForeignPtr Tensor
-> IO (ForeignPtr Tensor)
ATen.searchsorted_tsbbst) Tensor
_sorted_sequence Float
_self Bool
_out_int32 Bool
_right String
_side Tensor
_sorter

mse_loss ::
  -- | self
  Tensor ->
  -- | target
  Tensor ->
  -- | reduction
  Int ->
  Tensor
mse_loss :: Tensor -> Tensor -> Int -> Tensor
mse_loss Tensor
_self Tensor
_target Int
_reduction = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.mse_loss_ttl) Tensor
_self Tensor
_target Int
_reduction

l1_loss ::
  -- | self
  Tensor ->
  -- | target
  Tensor ->
  -- | reduction
  Int ->
  Tensor
l1_loss :: Tensor -> Tensor -> Int -> Tensor
l1_loss Tensor
_self Tensor
_target Int
_reduction = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.l1_loss_ttl) Tensor
_self Tensor
_target Int
_reduction

multi_margin_loss ::
  -- | self
  Tensor ->
  -- | target
  Tensor ->
  -- | p
  Float ->
  -- | margin
  Float ->
  -- | weight
  Tensor ->
  -- | reduction
  Int ->
  Tensor
multi_margin_loss :: Tensor -> Tensor -> Float -> Float -> Tensor -> Int -> Tensor
multi_margin_loss Tensor
_self Tensor
_target Float
_p Float
_margin Tensor
_weight Int
_reduction = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr Scalar
-> ForeignPtr Tensor
-> Int64
-> IO (ForeignPtr Tensor)
ATen.multi_margin_loss_ttsstl) Tensor
_self Tensor
_target Float
_p Float
_margin Tensor
_weight Int
_reduction

multilabel_margin_loss ::
  -- | self
  Tensor ->
  -- | target
  Tensor ->
  -- | reduction
  Int ->
  Tensor
multilabel_margin_loss :: Tensor -> Tensor -> Int -> Tensor
multilabel_margin_loss Tensor
_self Tensor
_target Int
_reduction = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.multilabel_margin_loss_ttl) Tensor
_self Tensor
_target Int
_reduction

multilabel_margin_loss_forward ::
  -- | self
  Tensor ->
  -- | target
  Tensor ->
  -- | reduction
  Int ->
  (Tensor, Tensor)
multilabel_margin_loss_forward :: Tensor -> Tensor -> Int -> (Tensor, Tensor)
multilabel_margin_loss_forward Tensor
_self Tensor
_target Int
_reduction = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor
-> Int64
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.multilabel_margin_loss_forward_ttl) Tensor
_self Tensor
_target Int
_reduction

nll_loss_nd ::
  -- | self
  Tensor ->
  -- | target
  Tensor ->
  -- | weight
  Tensor ->
  -- | reduction
  Int ->
  -- | ignore_index
  Int ->
  Tensor
nll_loss_nd :: Tensor -> Tensor -> Tensor -> Int -> Int -> Tensor
nll_loss_nd Tensor
_self Tensor
_target Tensor
_weight Int
_reduction Int
_ignore_index = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> Int64
-> Int64
-> IO (ForeignPtr Tensor)
ATen.nll_loss_nd_tttll) Tensor
_self Tensor
_target Tensor
_weight Int
_reduction Int
_ignore_index

nll_loss ::
  -- | self
  Tensor ->
  -- | target
  Tensor ->
  -- | weight
  Tensor ->
  -- | reduction
  Int ->
  -- | ignore_index
  Int ->
  Tensor
nll_loss :: Tensor -> Tensor -> Tensor -> Int -> Int -> Tensor
nll_loss Tensor
_self Tensor
_target Tensor
_weight Int
_reduction Int
_ignore_index = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> Int64
-> Int64
-> IO (ForeignPtr Tensor)
ATen.nll_loss_tttll) Tensor
_self Tensor
_target Tensor
_weight Int
_reduction Int
_ignore_index

nll_loss_forward ::
  -- | self
  Tensor ->
  -- | target
  Tensor ->
  -- | weight
  Tensor ->
  -- | reduction
  Int ->
  -- | ignore_index
  Int ->
  (Tensor, Tensor)
nll_loss_forward :: Tensor -> Tensor -> Tensor -> Int -> Int -> (Tensor, Tensor)
nll_loss_forward Tensor
_self Tensor
_target Tensor
_weight Int
_reduction Int
_ignore_index = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> Int64
-> Int64
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.nll_loss_forward_tttll) Tensor
_self Tensor
_target Tensor
_weight Int
_reduction Int
_ignore_index

nll_loss2d ::
  -- | self
  Tensor ->
  -- | target
  Tensor ->
  -- | weight
  Tensor ->
  -- | reduction
  Int ->
  -- | ignore_index
  Int ->
  Tensor
nll_loss2d :: Tensor -> Tensor -> Tensor -> Int -> Int -> Tensor
nll_loss2d Tensor
_self Tensor
_target Tensor
_weight Int
_reduction Int
_ignore_index = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> Int64
-> Int64
-> IO (ForeignPtr Tensor)
ATen.nll_loss2d_tttll) Tensor
_self Tensor
_target Tensor
_weight Int
_reduction Int
_ignore_index

nll_loss2d_forward ::
  -- | self
  Tensor ->
  -- | target
  Tensor ->
  -- | weight
  Tensor ->
  -- | reduction
  Int ->
  -- | ignore_index
  Int ->
  (Tensor, Tensor)
nll_loss2d_forward :: Tensor -> Tensor -> Tensor -> Int -> Int -> (Tensor, Tensor)
nll_loss2d_forward Tensor
_self Tensor
_target Tensor
_weight Int
_reduction Int
_ignore_index = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> Int64
-> Int64
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.nll_loss2d_forward_tttll) Tensor
_self Tensor
_target Tensor
_weight Int
_reduction Int
_ignore_index

smooth_l1_loss ::
  -- | self
  Tensor ->
  -- | target
  Tensor ->
  -- | reduction
  Int ->
  -- | beta
  Double ->
  Tensor
smooth_l1_loss :: Tensor -> Tensor -> Int -> Double -> Tensor
smooth_l1_loss Tensor
_self Tensor
_target Int
_reduction Double
_beta = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Tensor -> Int64 -> CDouble -> IO (ForeignPtr Tensor)
ATen.smooth_l1_loss_ttld) Tensor
_self Tensor
_target Int
_reduction Double
_beta

huber_loss ::
  -- | self
  Tensor ->
  -- | target
  Tensor ->
  -- | reduction
  Int ->
  -- | delta
  Double ->
  Tensor
huber_loss :: Tensor -> Tensor -> Int -> Double -> Tensor
huber_loss Tensor
_self Tensor
_target Int
_reduction Double
_delta = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Tensor -> Int64 -> CDouble -> IO (ForeignPtr Tensor)
ATen.huber_loss_ttld) Tensor
_self Tensor
_target Int
_reduction Double
_delta

soft_margin_loss ::
  -- | self
  Tensor ->
  -- | target
  Tensor ->
  -- | reduction
  Int ->
  Tensor
soft_margin_loss :: Tensor -> Tensor -> Int -> Tensor
soft_margin_loss Tensor
_self Tensor
_target Int
_reduction = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.soft_margin_loss_ttl) Tensor
_self Tensor
_target Int
_reduction

elu ::
  -- | self
  Tensor ->
  -- | alpha
  Float ->
  -- | scale
  Float ->
  -- | input_scale
  Float ->
  Tensor
elu :: Tensor -> Float -> Float -> Float -> Tensor
elu Tensor
_self Float
_alpha Float
_scale Float
_input_scale = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr Scalar
-> ForeignPtr Scalar
-> IO (ForeignPtr Tensor)
ATen.elu_tsss) Tensor
_self Float
_alpha Float
_scale Float
_input_scale

glu ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  Tensor
glu :: Tensor -> Int -> Tensor
glu Tensor
_self Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.glu_tl) Tensor
_self Int
_dim

hardsigmoid ::
  -- | self
  Tensor ->
  Tensor
hardsigmoid :: Tensor -> Tensor
hardsigmoid Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.hardsigmoid_t) Tensor
_self

hardtanh ::
  -- | self
  Tensor ->
  -- | min_val
  Float ->
  -- | max_val
  Float ->
  Tensor
hardtanh :: Tensor -> Float -> Float -> Tensor
hardtanh Tensor
_self Float
_min_val Float
_max_val = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Scalar -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.hardtanh_tss) Tensor
_self Float
_min_val Float
_max_val

hardswish ::
  -- | self
  Tensor ->
  Tensor
hardswish :: Tensor -> Tensor
hardswish Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.hardswish_t) Tensor
_self

leaky_relu ::
  -- | self
  Tensor ->
  -- | negative_slope
  Float ->
  Tensor
leaky_relu :: Tensor -> Float -> Tensor
leaky_relu Tensor
_self Float
_negative_slope = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.leaky_relu_ts) Tensor
_self Float
_negative_slope

log_sigmoid ::
  -- | self
  Tensor ->
  Tensor
log_sigmoid :: Tensor -> Tensor
log_sigmoid Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.log_sigmoid_t) Tensor
_self

log_sigmoid_forward ::
  -- | self
  Tensor ->
  (Tensor, Tensor)
log_sigmoid_forward :: Tensor -> (Tensor, Tensor)
log_sigmoid_forward Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.log_sigmoid_forward_t) Tensor
_self

softplus ::
  -- | self
  Tensor ->
  -- | beta
  Float ->
  -- | threshold
  Float ->
  Tensor
softplus :: Tensor -> Float -> Float -> Tensor
softplus Tensor
_self Float
_beta Float
_threshold = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Scalar -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.softplus_tss) Tensor
_self Float
_beta Float
_threshold

softshrink ::
  -- | self
  Tensor ->
  -- | lambd
  Float ->
  Tensor
softshrink :: Tensor -> Float -> Tensor
softshrink Tensor
_self Float
_lambd = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.softshrink_ts) Tensor
_self Float
_lambd

adaptive_avg_pool2d ::
  -- | self
  Tensor ->
  -- | output_size
  (Int, Int) ->
  Tensor
adaptive_avg_pool2d :: Tensor -> (Int, Int) -> Tensor
adaptive_avg_pool2d Tensor
_self (Int, Int)
_output_size = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr IntArray -> IO (ForeignPtr Tensor)
ATen.adaptive_avg_pool2d_tl) Tensor
_self (Int, Int)
_output_size

mkldnn_adaptive_avg_pool2d ::
  -- | self
  Tensor ->
  -- | output_size
  (Int, Int) ->
  Tensor
mkldnn_adaptive_avg_pool2d :: Tensor -> (Int, Int) -> Tensor
mkldnn_adaptive_avg_pool2d Tensor
_self (Int, Int)
_output_size = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr IntArray -> IO (ForeignPtr Tensor)
ATen.mkldnn_adaptive_avg_pool2d_tl) Tensor
_self (Int, Int)
_output_size

adaptive_avg_pool3d ::
  -- | self
  Tensor ->
  -- | output_size
  (Int, Int, Int) ->
  Tensor
adaptive_avg_pool3d :: Tensor -> (Int, Int, Int) -> Tensor
adaptive_avg_pool3d Tensor
_self (Int, Int, Int)
_output_size = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr IntArray -> IO (ForeignPtr Tensor)
ATen.adaptive_avg_pool3d_tl) Tensor
_self (Int, Int, Int)
_output_size

adaptive_max_pool2d ::
  -- | self
  Tensor ->
  -- | output_size
  (Int, Int) ->
  (Tensor, Tensor)
adaptive_max_pool2d :: Tensor -> (Int, Int) -> (Tensor, Tensor)
adaptive_max_pool2d Tensor
_self (Int, Int)
_output_size = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor
-> ForeignPtr IntArray
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.adaptive_max_pool2d_tl) Tensor
_self (Int, Int)
_output_size

adaptive_max_pool3d ::
  -- | self
  Tensor ->
  -- | output_size
  (Int, Int, Int) ->
  (Tensor, Tensor)
adaptive_max_pool3d :: Tensor -> (Int, Int, Int) -> (Tensor, Tensor)
adaptive_max_pool3d Tensor
_self (Int, Int, Int)
_output_size = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor
-> ForeignPtr IntArray
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.adaptive_max_pool3d_tl) Tensor
_self (Int, Int, Int)
_output_size

avg_pool2d ::
  -- | self
  Tensor ->
  -- | kernel_size
  (Int, Int) ->
  -- | stride
  (Int, Int) ->
  -- | padding
  (Int, Int) ->
  -- | ceil_mode
  Bool ->
  -- | count_include_pad
  Bool ->
  -- | divisor_override
  Int ->
  Tensor
avg_pool2d :: Tensor
-> (Int, Int)
-> (Int, Int)
-> (Int, Int)
-> Bool
-> Bool
-> Int
-> Tensor
avg_pool2d Tensor
_self (Int, Int)
_kernel_size (Int, Int)
_stride (Int, Int)
_padding Bool
_ceil_mode Bool
_count_include_pad Int
_divisor_override = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> IO y
cast7 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> CBool
-> CBool
-> Int64
-> IO (ForeignPtr Tensor)
ATen.avg_pool2d_tlllbbl) Tensor
_self (Int, Int)
_kernel_size (Int, Int)
_stride (Int, Int)
_padding Bool
_ceil_mode Bool
_count_include_pad Int
_divisor_override

avg_pool3d ::
  -- | self
  Tensor ->
  -- | kernel_size
  (Int, Int, Int) ->
  -- | stride
  (Int, Int, Int) ->
  -- | padding
  (Int, Int, Int) ->
  -- | ceil_mode
  Bool ->
  -- | count_include_pad
  Bool ->
  -- | divisor_override
  Int ->
  Tensor
avg_pool3d :: Tensor
-> (Int, Int, Int)
-> (Int, Int, Int)
-> (Int, Int, Int)
-> Bool
-> Bool
-> Int
-> Tensor
avg_pool3d Tensor
_self (Int, Int, Int)
_kernel_size (Int, Int, Int)
_stride (Int, Int, Int)
_padding Bool
_ceil_mode Bool
_count_include_pad Int
_divisor_override = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> IO y
cast7 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> CBool
-> CBool
-> Int64
-> IO (ForeignPtr Tensor)
ATen.avg_pool3d_tlllbbl) Tensor
_self (Int, Int, Int)
_kernel_size (Int, Int, Int)
_stride (Int, Int, Int)
_padding Bool
_ceil_mode Bool
_count_include_pad Int
_divisor_override

fractional_max_pool2d ::
  -- | self
  Tensor ->
  -- | kernel_size
  (Int, Int) ->
  -- | output_size
  (Int, Int) ->
  -- | random_samples
  Tensor ->
  (Tensor, Tensor)
fractional_max_pool2d :: Tensor -> (Int, Int) -> (Int, Int) -> Tensor -> (Tensor, Tensor)
fractional_max_pool2d Tensor
_self (Int, Int)
_kernel_size (Int, Int)
_output_size Tensor
_random_samples = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr Tensor
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.fractional_max_pool2d_tllt) Tensor
_self (Int, Int)
_kernel_size (Int, Int)
_output_size Tensor
_random_samples

fractional_max_pool3d ::
  -- | self
  Tensor ->
  -- | kernel_size
  (Int, Int, Int) ->
  -- | output_size
  (Int, Int, Int) ->
  -- | random_samples
  Tensor ->
  (Tensor, Tensor)
fractional_max_pool3d :: Tensor
-> (Int, Int, Int) -> (Int, Int, Int) -> Tensor -> (Tensor, Tensor)
fractional_max_pool3d Tensor
_self (Int, Int, Int)
_kernel_size (Int, Int, Int)
_output_size Tensor
_random_samples = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr Tensor
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.fractional_max_pool3d_tllt) Tensor
_self (Int, Int, Int)
_kernel_size (Int, Int, Int)
_output_size Tensor
_random_samples

max_pool2d_with_indices ::
  -- | self
  Tensor ->
  -- | kernel_size
  (Int, Int) ->
  -- | stride
  (Int, Int) ->
  -- | padding
  (Int, Int) ->
  -- | dilation
  (Int, Int) ->
  -- | ceil_mode
  Bool ->
  (Tensor, Tensor)
max_pool2d_with_indices :: Tensor
-> (Int, Int)
-> (Int, Int)
-> (Int, Int)
-> (Int, Int)
-> Bool
-> (Tensor, Tensor)
max_pool2d_with_indices Tensor
_self (Int, Int)
_kernel_size (Int, Int)
_stride (Int, Int)
_padding (Int, Int)
_dilation Bool
_ceil_mode = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.max_pool2d_with_indices_tllllb) Tensor
_self (Int, Int)
_kernel_size (Int, Int)
_stride (Int, Int)
_padding (Int, Int)
_dilation Bool
_ceil_mode

max_pool3d_with_indices ::
  -- | self
  Tensor ->
  -- | kernel_size
  (Int, Int, Int) ->
  -- | stride
  (Int, Int, Int) ->
  -- | padding
  (Int, Int, Int) ->
  -- | dilation
  (Int, Int, Int) ->
  -- | ceil_mode
  Bool ->
  (Tensor, Tensor)
max_pool3d_with_indices :: Tensor
-> (Int, Int, Int)
-> (Int, Int, Int)
-> (Int, Int, Int)
-> (Int, Int, Int)
-> Bool
-> (Tensor, Tensor)
max_pool3d_with_indices Tensor
_self (Int, Int, Int)
_kernel_size (Int, Int, Int)
_stride (Int, Int, Int)
_padding (Int, Int, Int)
_dilation Bool
_ceil_mode = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.max_pool3d_with_indices_tllllb) Tensor
_self (Int, Int, Int)
_kernel_size (Int, Int, Int)
_stride (Int, Int, Int)
_padding (Int, Int, Int)
_dilation Bool
_ceil_mode

max_unpool2d ::
  -- | self
  Tensor ->
  -- | indices
  Tensor ->
  -- | output_size
  (Int, Int) ->
  Tensor
max_unpool2d :: Tensor -> Tensor -> (Int, Int) -> Tensor
max_unpool2d Tensor
_self Tensor
_indices (Int, Int)
_output_size = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> IO (ForeignPtr Tensor)
ATen.max_unpool2d_ttl) Tensor
_self Tensor
_indices (Int, Int)
_output_size

max_unpool3d ::
  -- | self
  Tensor ->
  -- | indices
  Tensor ->
  -- | output_size
  (Int, Int, Int) ->
  -- | stride
  (Int, Int, Int) ->
  -- | padding
  (Int, Int, Int) ->
  Tensor
max_unpool3d :: Tensor
-> Tensor
-> (Int, Int, Int)
-> (Int, Int, Int)
-> (Int, Int, Int)
-> Tensor
max_unpool3d Tensor
_self Tensor
_indices (Int, Int, Int)
_output_size (Int, Int, Int)
_stride (Int, Int, Int)
_padding = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> IO (ForeignPtr Tensor)
ATen.max_unpool3d_ttlll) Tensor
_self Tensor
_indices (Int, Int, Int)
_output_size (Int, Int, Int)
_stride (Int, Int, Int)
_padding

reflection_pad1d ::
  -- | self
  Tensor ->
  -- | padding
  (Int, Int) ->
  Tensor
reflection_pad1d :: Tensor -> (Int, Int) -> Tensor
reflection_pad1d Tensor
_self (Int, Int)
_padding = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr IntArray -> IO (ForeignPtr Tensor)
ATen.reflection_pad1d_tl) Tensor
_self (Int, Int)
_padding

reflection_pad2d ::
  -- | self
  Tensor ->
  -- | padding
  (Int, Int, Int, Int) ->
  Tensor
reflection_pad2d :: Tensor -> (Int, Int, Int, Int) -> Tensor
reflection_pad2d Tensor
_self (Int, Int, Int, Int)
_padding = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr IntArray -> IO (ForeignPtr Tensor)
ATen.reflection_pad2d_tl) Tensor
_self (Int, Int, Int, Int)
_padding

reflection_pad3d ::
  -- | self
  Tensor ->
  -- | padding
  (Int, Int, Int, Int, Int, Int) ->
  Tensor
reflection_pad3d :: Tensor -> (Int, Int, Int, Int, Int, Int) -> Tensor
reflection_pad3d Tensor
_self (Int, Int, Int, Int, Int, Int)
_padding = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr IntArray -> IO (ForeignPtr Tensor)
ATen.reflection_pad3d_tl) Tensor
_self (Int, Int, Int, Int, Int, Int)
_padding

replication_pad1d ::
  -- | self
  Tensor ->
  -- | padding
  (Int, Int) ->
  Tensor
replication_pad1d :: Tensor -> (Int, Int) -> Tensor
replication_pad1d Tensor
_self (Int, Int)
_padding = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr IntArray -> IO (ForeignPtr Tensor)
ATen.replication_pad1d_tl) Tensor
_self (Int, Int)
_padding

replication_pad2d ::
  -- | self
  Tensor ->
  -- | padding
  (Int, Int, Int, Int) ->
  Tensor
replication_pad2d :: Tensor -> (Int, Int, Int, Int) -> Tensor
replication_pad2d Tensor
_self (Int, Int, Int, Int)
_padding = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr IntArray -> IO (ForeignPtr Tensor)
ATen.replication_pad2d_tl) Tensor
_self (Int, Int, Int, Int)
_padding

replication_pad3d ::
  -- | self
  Tensor ->
  -- | padding
  (Int, Int, Int, Int, Int, Int) ->
  Tensor
replication_pad3d :: Tensor -> (Int, Int, Int, Int, Int, Int) -> Tensor
replication_pad3d Tensor
_self (Int, Int, Int, Int, Int, Int)
_padding = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr IntArray -> IO (ForeignPtr Tensor)
ATen.replication_pad3d_tl) Tensor
_self (Int, Int, Int, Int, Int, Int)
_padding

upsample_linear1d_tlba ::
  -- | input
  Tensor ->
  -- | output_size
  [Int] ->
  -- | align_corners
  Bool ->
  -- | scale_factors
  ([Double]) ->
  Tensor
upsample_linear1d_tlba :: Tensor -> [Int] -> Bool -> [Double] -> Tensor
upsample_linear1d_tlba Tensor
_input [Int]
_output_size Bool
_align_corners [Double]
_scale_factors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> CBool
-> ForeignPtr (StdVector CDouble)
-> IO (ForeignPtr Tensor)
ATen.upsample_linear1d_tlba) Tensor
_input [Int]
_output_size Bool
_align_corners [Double]
_scale_factors

upsample_bilinear2d_tlba ::
  -- | input
  Tensor ->
  -- | output_size
  [Int] ->
  -- | align_corners
  Bool ->
  -- | scale_factors
  ([Double]) ->
  Tensor
upsample_bilinear2d_tlba :: Tensor -> [Int] -> Bool -> [Double] -> Tensor
upsample_bilinear2d_tlba Tensor
_input [Int]
_output_size Bool
_align_corners [Double]
_scale_factors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> CBool
-> ForeignPtr (StdVector CDouble)
-> IO (ForeignPtr Tensor)
ATen.upsample_bilinear2d_tlba) Tensor
_input [Int]
_output_size Bool
_align_corners [Double]
_scale_factors

upsample_trilinear3d_tlba ::
  -- | input
  Tensor ->
  -- | output_size
  [Int] ->
  -- | align_corners
  Bool ->
  -- | scale_factors
  ([Double]) ->
  Tensor
upsample_trilinear3d_tlba :: Tensor -> [Int] -> Bool -> [Double] -> Tensor
upsample_trilinear3d_tlba Tensor
_input [Int]
_output_size Bool
_align_corners [Double]
_scale_factors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> CBool
-> ForeignPtr (StdVector CDouble)
-> IO (ForeignPtr Tensor)
ATen.upsample_trilinear3d_tlba) Tensor
_input [Int]
_output_size Bool
_align_corners [Double]
_scale_factors

upsample_bicubic2d_tlba ::
  -- | input
  Tensor ->
  -- | output_size
  [Int] ->
  -- | align_corners
  Bool ->
  -- | scale_factors
  ([Double]) ->
  Tensor
upsample_bicubic2d_tlba :: Tensor -> [Int] -> Bool -> [Double] -> Tensor
upsample_bicubic2d_tlba Tensor
_input [Int]
_output_size Bool
_align_corners [Double]
_scale_factors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> CBool
-> ForeignPtr (StdVector CDouble)
-> IO (ForeignPtr Tensor)
ATen.upsample_bicubic2d_tlba) Tensor
_input [Int]
_output_size Bool
_align_corners [Double]
_scale_factors

upsample_nearest1d_tla ::
  -- | input
  Tensor ->
  -- | output_size
  [Int] ->
  -- | scale_factors
  ([Double]) ->
  Tensor
upsample_nearest1d_tla :: Tensor -> [Int] -> [Double] -> Tensor
upsample_nearest1d_tla Tensor
_input [Int]
_output_size [Double]
_scale_factors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr (StdVector CDouble)
-> IO (ForeignPtr Tensor)
ATen.upsample_nearest1d_tla) Tensor
_input [Int]
_output_size [Double]
_scale_factors

upsample_nearest2d_tla ::
  -- | input
  Tensor ->
  -- | output_size
  [Int] ->
  -- | scale_factors
  ([Double]) ->
  Tensor
upsample_nearest2d_tla :: Tensor -> [Int] -> [Double] -> Tensor
upsample_nearest2d_tla Tensor
_input [Int]
_output_size [Double]
_scale_factors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr (StdVector CDouble)
-> IO (ForeignPtr Tensor)
ATen.upsample_nearest2d_tla) Tensor
_input [Int]
_output_size [Double]
_scale_factors

upsample_nearest3d_tla ::
  -- | input
  Tensor ->
  -- | output_size
  [Int] ->
  -- | scale_factors
  ([Double]) ->
  Tensor
upsample_nearest3d_tla :: Tensor -> [Int] -> [Double] -> Tensor
upsample_nearest3d_tla Tensor
_input [Int]
_output_size [Double]
_scale_factors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr (StdVector CDouble)
-> IO (ForeignPtr Tensor)
ATen.upsample_nearest3d_tla) Tensor
_input [Int]
_output_size [Double]
_scale_factors

-- upsample_linear1d_tlbd
--   :: Tensor -- ^ self
--   -> Int -- ^ output_size
--   -> Bool -- ^ align_corners
--   -> Double -- ^ scales
--   -> Tensor
-- upsample_linear1d_tlbd _self _output_size _align_corners _scales = unsafePerformIO $ (cast4 ATen.upsample_linear1d_tlbd) _self _output_size _align_corners _scales

upsample_bilinear2d_tlbdd ::
  -- | self
  Tensor ->
  -- | output_size
  (Int, Int) ->
  -- | align_corners
  Bool ->
  -- | scales_h
  Double ->
  -- | scales_w
  Double ->
  Tensor
upsample_bilinear2d_tlbdd :: Tensor -> (Int, Int) -> Bool -> Double -> Double -> Tensor
upsample_bilinear2d_tlbdd Tensor
_self (Int, Int)
_output_size Bool
_align_corners Double
_scales_h Double
_scales_w = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr IntArray
-> CBool
-> CDouble
-> CDouble
-> IO (ForeignPtr Tensor)
ATen.upsample_bilinear2d_tlbdd) Tensor
_self (Int, Int)
_output_size Bool
_align_corners Double
_scales_h Double
_scales_w

upsample_bicubic2d_tlbdd ::
  -- | self
  Tensor ->
  -- | output_size
  (Int, Int) ->
  -- | align_corners
  Bool ->
  -- | scales_h
  Double ->
  -- | scales_w
  Double ->
  Tensor
upsample_bicubic2d_tlbdd :: Tensor -> (Int, Int) -> Bool -> Double -> Double -> Tensor
upsample_bicubic2d_tlbdd Tensor
_self (Int, Int)
_output_size Bool
_align_corners Double
_scales_h Double
_scales_w = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr IntArray
-> CBool
-> CDouble
-> CDouble
-> IO (ForeignPtr Tensor)
ATen.upsample_bicubic2d_tlbdd) Tensor
_self (Int, Int)
_output_size Bool
_align_corners Double
_scales_h Double
_scales_w

upsample_trilinear3d_tlbddd ::
  -- | self
  Tensor ->
  -- | output_size
  (Int, Int, Int) ->
  -- | align_corners
  Bool ->
  -- | scales_d
  Double ->
  -- | scales_h
  Double ->
  -- | scales_w
  Double ->
  Tensor
upsample_trilinear3d_tlbddd :: Tensor
-> (Int, Int, Int) -> Bool -> Double -> Double -> Double -> Tensor
upsample_trilinear3d_tlbddd Tensor
_self (Int, Int, Int)
_output_size Bool
_align_corners Double
_scales_d Double
_scales_h Double
_scales_w = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr IntArray
-> CBool
-> CDouble
-> CDouble
-> CDouble
-> IO (ForeignPtr Tensor)
ATen.upsample_trilinear3d_tlbddd) Tensor
_self (Int, Int, Int)
_output_size Bool
_align_corners Double
_scales_d Double
_scales_h Double
_scales_w

-- upsample_nearest1d_tld
--   :: Tensor -- ^ self
--   -> Int -- ^ output_size
--   -> Double -- ^ scales
--   -> Tensor
-- upsample_nearest1d_tld _self _output_size _scales = unsafePerformIO $ (cast3 ATen.upsample_nearest1d_tld) _self _output_size _scales

upsample_nearest2d_tldd ::
  -- | self
  Tensor ->
  -- | output_size
  (Int, Int) ->
  -- | scales_h
  Double ->
  -- | scales_w
  Double ->
  Tensor
upsample_nearest2d_tldd :: Tensor -> (Int, Int) -> Double -> Double -> Tensor
upsample_nearest2d_tldd Tensor
_self (Int, Int)
_output_size Double
_scales_h Double
_scales_w = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> CDouble
-> CDouble
-> IO (ForeignPtr Tensor)
ATen.upsample_nearest2d_tldd) Tensor
_self (Int, Int)
_output_size Double
_scales_h Double
_scales_w

upsample_nearest3d_tlddd ::
  -- | self
  Tensor ->
  -- | output_size
  (Int, Int, Int) ->
  -- | scales_d
  Double ->
  -- | scales_h
  Double ->
  -- | scales_w
  Double ->
  Tensor
upsample_nearest3d_tlddd :: Tensor -> (Int, Int, Int) -> Double -> Double -> Double -> Tensor
upsample_nearest3d_tlddd Tensor
_self (Int, Int, Int)
_output_size Double
_scales_d Double
_scales_h Double
_scales_w = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr IntArray
-> CDouble
-> CDouble
-> CDouble
-> IO (ForeignPtr Tensor)
ATen.upsample_nearest3d_tlddd) Tensor
_self (Int, Int, Int)
_output_size Double
_scales_d Double
_scales_h Double
_scales_w

slow_conv_transpose2d ::
  -- | self
  Tensor ->
  -- | weight
  Tensor ->
  -- | kernel_size
  (Int, Int) ->
  -- | bias
  Tensor ->
  -- | stride
  (Int, Int) ->
  -- | padding
  (Int, Int) ->
  -- | output_padding
  (Int, Int) ->
  -- | dilation
  (Int, Int) ->
  Tensor
slow_conv_transpose2d :: Tensor
-> Tensor
-> (Int, Int)
-> Tensor
-> (Int, Int)
-> (Int, Int)
-> (Int, Int)
-> (Int, Int)
-> Tensor
slow_conv_transpose2d Tensor
_self Tensor
_weight (Int, Int)
_kernel_size Tensor
_bias (Int, Int)
_stride (Int, Int)
_padding (Int, Int)
_output_padding (Int, Int)
_dilation = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> IO y
cast8 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> IO (ForeignPtr Tensor)
ATen.slow_conv_transpose2d_ttltllll) Tensor
_self Tensor
_weight (Int, Int)
_kernel_size Tensor
_bias (Int, Int)
_stride (Int, Int)
_padding (Int, Int)
_output_padding (Int, Int)
_dilation

slow_conv_transpose3d ::
  -- | self
  Tensor ->
  -- | weight
  Tensor ->
  -- | kernel_size
  (Int, Int, Int) ->
  -- | bias
  Tensor ->
  -- | stride
  (Int, Int, Int) ->
  -- | padding
  (Int, Int, Int) ->
  -- | output_padding
  (Int, Int, Int) ->
  -- | dilation
  (Int, Int, Int) ->
  Tensor
slow_conv_transpose3d :: Tensor
-> Tensor
-> (Int, Int, Int)
-> Tensor
-> (Int, Int, Int)
-> (Int, Int, Int)
-> (Int, Int, Int)
-> (Int, Int, Int)
-> Tensor
slow_conv_transpose3d Tensor
_self Tensor
_weight (Int, Int, Int)
_kernel_size Tensor
_bias (Int, Int, Int)
_stride (Int, Int, Int)
_padding (Int, Int, Int)
_output_padding (Int, Int, Int)
_dilation = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 x7 cx7 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6, Castable x7 cx7,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> cx7 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> x7 -> IO y
cast8 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> IO (ForeignPtr Tensor)
ATen.slow_conv_transpose3d_ttltllll) Tensor
_self Tensor
_weight (Int, Int, Int)
_kernel_size Tensor
_bias (Int, Int, Int)
_stride (Int, Int, Int)
_padding (Int, Int, Int)
_output_padding (Int, Int, Int)
_dilation

thnn_conv2d ::
  -- | self
  Tensor ->
  -- | weight
  Tensor ->
  -- | kernel_size
  (Int, Int) ->
  -- | bias
  Tensor ->
  -- | stride
  (Int, Int) ->
  -- | padding
  (Int, Int) ->
  Tensor
thnn_conv2d :: Tensor
-> Tensor
-> (Int, Int)
-> Tensor
-> (Int, Int)
-> (Int, Int)
-> Tensor
thnn_conv2d Tensor
_self Tensor
_weight (Int, Int)
_kernel_size Tensor
_bias (Int, Int)
_stride (Int, Int)
_padding = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> IO (ForeignPtr Tensor)
ATen.thnn_conv2d_ttltll) Tensor
_self Tensor
_weight (Int, Int)
_kernel_size Tensor
_bias (Int, Int)
_stride (Int, Int)
_padding

conv_depthwise3d ::
  -- | self
  Tensor ->
  -- | weight
  Tensor ->
  -- | kernel_size
  (Int, Int, Int) ->
  -- | bias
  Tensor ->
  -- | stride
  (Int, Int, Int) ->
  -- | padding
  (Int, Int, Int) ->
  -- | dilation
  (Int, Int, Int) ->
  Tensor
conv_depthwise3d :: Tensor
-> Tensor
-> (Int, Int, Int)
-> Tensor
-> (Int, Int, Int)
-> (Int, Int, Int)
-> (Int, Int, Int)
-> Tensor
conv_depthwise3d Tensor
_self Tensor
_weight (Int, Int, Int)
_kernel_size Tensor
_bias (Int, Int, Int)
_stride (Int, Int, Int)
_padding (Int, Int, Int)
_dilation = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> IO y
cast7 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> IO (ForeignPtr Tensor)
ATen.conv_depthwise3d_ttltlll) Tensor
_self Tensor
_weight (Int, Int, Int)
_kernel_size Tensor
_bias (Int, Int, Int)
_stride (Int, Int, Int)
_padding (Int, Int, Int)
_dilation

slow_conv3d ::
  -- | self
  Tensor ->
  -- | weight
  Tensor ->
  -- | kernel_size
  (Int, Int, Int) ->
  -- | bias
  Tensor ->
  -- | stride
  (Int, Int, Int) ->
  -- | padding
  (Int, Int, Int) ->
  Tensor
slow_conv3d :: Tensor
-> Tensor
-> (Int, Int, Int)
-> Tensor
-> (Int, Int, Int)
-> (Int, Int, Int)
-> Tensor
slow_conv3d Tensor
_self Tensor
_weight (Int, Int, Int)
_kernel_size Tensor
_bias (Int, Int, Int)
_stride (Int, Int, Int)
_padding = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> IO (ForeignPtr Tensor)
ATen.slow_conv3d_ttltll) Tensor
_self Tensor
_weight (Int, Int, Int)
_kernel_size Tensor
_bias (Int, Int, Int)
_stride (Int, Int, Int)
_padding

slow_conv3d_forward ::
  -- | self
  Tensor ->
  -- | weight
  Tensor ->
  -- | kernel_size
  (Int, Int, Int) ->
  -- | bias
  Tensor ->
  -- | stride
  (Int, Int, Int) ->
  -- | padding
  (Int, Int, Int) ->
  Tensor
slow_conv3d_forward :: Tensor
-> Tensor
-> (Int, Int, Int)
-> Tensor
-> (Int, Int, Int)
-> (Int, Int, Int)
-> Tensor
slow_conv3d_forward Tensor
_self Tensor
_weight (Int, Int, Int)
_kernel_size Tensor
_bias (Int, Int, Int)
_stride (Int, Int, Int)
_padding = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> IO (ForeignPtr Tensor)
ATen.slow_conv3d_forward_ttltll) Tensor
_self Tensor
_weight (Int, Int, Int)
_kernel_size Tensor
_bias (Int, Int, Int)
_stride (Int, Int, Int)
_padding

slow_conv_dilated2d ::
  -- | self
  Tensor ->
  -- | weight
  Tensor ->
  -- | kernel_size
  (Int, Int) ->
  -- | bias
  Tensor ->
  -- | stride
  (Int, Int) ->
  -- | padding
  (Int, Int) ->
  -- | dilation
  (Int, Int) ->
  Tensor
slow_conv_dilated2d :: Tensor
-> Tensor
-> (Int, Int)
-> Tensor
-> (Int, Int)
-> (Int, Int)
-> (Int, Int)
-> Tensor
slow_conv_dilated2d Tensor
_self Tensor
_weight (Int, Int)
_kernel_size Tensor
_bias (Int, Int)
_stride (Int, Int)
_padding (Int, Int)
_dilation = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> IO y
cast7 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> IO (ForeignPtr Tensor)
ATen.slow_conv_dilated2d_ttltlll) Tensor
_self Tensor
_weight (Int, Int)
_kernel_size Tensor
_bias (Int, Int)
_stride (Int, Int)
_padding (Int, Int)
_dilation

slow_conv_dilated3d ::
  -- | self
  Tensor ->
  -- | weight
  Tensor ->
  -- | kernel_size
  (Int, Int, Int) ->
  -- | bias
  Tensor ->
  -- | stride
  (Int, Int, Int) ->
  -- | padding
  (Int, Int, Int) ->
  -- | dilation
  (Int, Int, Int) ->
  Tensor
slow_conv_dilated3d :: Tensor
-> Tensor
-> (Int, Int, Int)
-> Tensor
-> (Int, Int, Int)
-> (Int, Int, Int)
-> (Int, Int, Int)
-> Tensor
slow_conv_dilated3d Tensor
_self Tensor
_weight (Int, Int, Int)
_kernel_size Tensor
_bias (Int, Int, Int)
_stride (Int, Int, Int)
_padding (Int, Int, Int)
_dilation = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> IO y
cast7 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> IO (ForeignPtr Tensor)
ATen.slow_conv_dilated3d_ttltlll) Tensor
_self Tensor
_weight (Int, Int, Int)
_kernel_size Tensor
_bias (Int, Int, Int)
_stride (Int, Int, Int)
_padding (Int, Int, Int)
_dilation

col2im ::
  -- | self
  Tensor ->
  -- | output_size
  (Int, Int) ->
  -- | kernel_size
  (Int, Int) ->
  -- | dilation
  (Int, Int) ->
  -- | padding
  (Int, Int) ->
  -- | stride
  (Int, Int) ->
  Tensor
col2im :: Tensor
-> (Int, Int)
-> (Int, Int)
-> (Int, Int)
-> (Int, Int)
-> (Int, Int)
-> Tensor
col2im Tensor
_self (Int, Int)
_output_size (Int, Int)
_kernel_size (Int, Int)
_dilation (Int, Int)
_padding (Int, Int)
_stride = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> IO y
cast6 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> IO (ForeignPtr Tensor)
ATen.col2im_tlllll) Tensor
_self (Int, Int)
_output_size (Int, Int)
_kernel_size (Int, Int)
_dilation (Int, Int)
_padding (Int, Int)
_stride

column_stack ::
  -- | tensors
  [Tensor] ->
  Tensor
column_stack :: [Tensor] -> Tensor
column_stack [Tensor]
_tensors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr TensorList -> IO (ForeignPtr Tensor)
ATen.column_stack_l) [Tensor]
_tensors

im2col ::
  -- | self
  Tensor ->
  -- | kernel_size
  (Int, Int) ->
  -- | dilation
  (Int, Int) ->
  -- | padding
  (Int, Int) ->
  -- | stride
  (Int, Int) ->
  Tensor
im2col :: Tensor
-> (Int, Int) -> (Int, Int) -> (Int, Int) -> (Int, Int) -> Tensor
im2col Tensor
_self (Int, Int)
_kernel_size (Int, Int)
_dilation (Int, Int)
_padding (Int, Int)
_stride = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> IO (ForeignPtr Tensor)
ATen.im2col_tllll) Tensor
_self (Int, Int)
_kernel_size (Int, Int)
_dilation (Int, Int)
_padding (Int, Int)
_stride

isfinite ::
  -- | self
  Tensor ->
  Tensor
isfinite :: Tensor -> Tensor
isfinite Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.isfinite_t) Tensor
_self

isinf ::
  -- | self
  Tensor ->
  Tensor
isinf :: Tensor -> Tensor
isinf Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.isinf_t) Tensor
_self

isposinf ::
  -- | self
  Tensor ->
  Tensor
isposinf :: Tensor -> Tensor
isposinf Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.isposinf_t) Tensor
_self

isneginf ::
  -- | self
  Tensor ->
  Tensor
isneginf :: Tensor -> Tensor
isneginf Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.isneginf_t) Tensor
_self

special_entr ::
  -- | self
  Tensor ->
  Tensor
special_entr :: Tensor -> Tensor
special_entr Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_entr_t) Tensor
_self

special_ndtri ::
  -- | self
  Tensor ->
  Tensor
special_ndtri :: Tensor -> Tensor
special_ndtri Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_ndtri_t) Tensor
_self

special_expm1 ::
  -- | self
  Tensor ->
  Tensor
special_expm1 :: Tensor -> Tensor
special_expm1 Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_expm1_t) Tensor
_self

special_exp2 ::
  -- | self
  Tensor ->
  Tensor
special_exp2 :: Tensor -> Tensor
special_exp2 Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_exp2_t) Tensor
_self

special_psi ::
  -- | self
  Tensor ->
  Tensor
special_psi :: Tensor -> Tensor
special_psi Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_psi_t) Tensor
_self

special_digamma ::
  -- | self
  Tensor ->
  Tensor
special_digamma :: Tensor -> Tensor
special_digamma Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_digamma_t) Tensor
_self

special_gammaln ::
  -- | self
  Tensor ->
  Tensor
special_gammaln :: Tensor -> Tensor
special_gammaln Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_gammaln_t) Tensor
_self

special_erf ::
  -- | self
  Tensor ->
  Tensor
special_erf :: Tensor -> Tensor
special_erf Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_erf_t) Tensor
_self

special_erfc ::
  -- | self
  Tensor ->
  Tensor
special_erfc :: Tensor -> Tensor
special_erfc Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_erfc_t) Tensor
_self

special_erfcx ::
  -- | self
  Tensor ->
  Tensor
special_erfcx :: Tensor -> Tensor
special_erfcx Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_erfcx_t) Tensor
_self

special_erfinv ::
  -- | self
  Tensor ->
  Tensor
special_erfinv :: Tensor -> Tensor
special_erfinv Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_erfinv_t) Tensor
_self

special_ndtr ::
  -- | self
  Tensor ->
  Tensor
special_ndtr :: Tensor -> Tensor
special_ndtr Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_ndtr_t) Tensor
_self

special_xlog1py_tt ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
special_xlog1py_tt :: Tensor -> Tensor -> Tensor
special_xlog1py_tt Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_xlog1py_tt) Tensor
_self Tensor
_other

special_xlog1py_st ::
  -- | self
  Float ->
  -- | other
  Tensor ->
  Tensor
special_xlog1py_st :: Float -> Tensor -> Tensor
special_xlog1py_st Float
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Scalar -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_xlog1py_st) Float
_self Tensor
_other

special_xlog1py_ts ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
special_xlog1py_ts :: Tensor -> Float -> Tensor
special_xlog1py_ts Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.special_xlog1py_ts) Tensor
_self Float
_other

special_xlogy_tt ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
special_xlogy_tt :: Tensor -> Tensor -> Tensor
special_xlogy_tt Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_xlogy_tt) Tensor
_self Tensor
_other

special_xlogy_st ::
  -- | self
  Float ->
  -- | other
  Tensor ->
  Tensor
special_xlogy_st :: Float -> Tensor -> Tensor
special_xlogy_st Float
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Scalar -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_xlogy_st) Float
_self Tensor
_other

special_xlogy_ts ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
special_xlogy_ts :: Tensor -> Float -> Tensor
special_xlogy_ts Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.special_xlogy_ts) Tensor
_self Float
_other

special_zeta_tt ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
special_zeta_tt :: Tensor -> Tensor -> Tensor
special_zeta_tt Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_zeta_tt) Tensor
_self Tensor
_other

special_zeta_st ::
  -- | self
  Float ->
  -- | other
  Tensor ->
  Tensor
special_zeta_st :: Float -> Tensor -> Tensor
special_zeta_st Float
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Scalar -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_zeta_st) Float
_self Tensor
_other

special_zeta_ts ::
  -- | self
  Tensor ->
  -- | other
  Float ->
  Tensor
special_zeta_ts :: Tensor -> Float -> Tensor
special_zeta_ts Tensor
_self Float
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.special_zeta_ts) Tensor
_self Float
_other

special_i0 ::
  -- | self
  Tensor ->
  Tensor
special_i0 :: Tensor -> Tensor
special_i0 Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_i0_t) Tensor
_self

special_i0e ::
  -- | self
  Tensor ->
  Tensor
special_i0e :: Tensor -> Tensor
special_i0e Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_i0e_t) Tensor
_self

special_i1 ::
  -- | self
  Tensor ->
  Tensor
special_i1 :: Tensor -> Tensor
special_i1 Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_i1_t) Tensor
_self

special_i1e ::
  -- | self
  Tensor ->
  Tensor
special_i1e :: Tensor -> Tensor
special_i1e Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_i1e_t) Tensor
_self

special_logit ::
  -- | self
  Tensor ->
  -- | eps
  Double ->
  Tensor
special_logit :: Tensor -> Double -> Tensor
special_logit Tensor
_self Double
_eps = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> CDouble -> IO (ForeignPtr Tensor)
ATen.special_logit_td) Tensor
_self Double
_eps

special_polygamma ::
  -- | n
  Int ->
  -- | self
  Tensor ->
  Tensor
special_polygamma :: Int -> Tensor -> Tensor
special_polygamma Int
_n Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 Int64 -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_polygamma_lt) Int
_n Tensor
_self

special_logsumexp ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  Tensor
special_logsumexp :: Tensor -> Int -> Bool -> Tensor
special_logsumexp Tensor
_self Int
_dim Bool
_keepdim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr IntArray -> CBool -> IO (ForeignPtr Tensor)
ATen.special_logsumexp_tlb) Tensor
_self Int
_dim Bool
_keepdim

special_expit ::
  -- | self
  Tensor ->
  Tensor
special_expit :: Tensor -> Tensor
special_expit Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_expit_t) Tensor
_self

special_sinc ::
  -- | self
  Tensor ->
  Tensor
special_sinc :: Tensor -> Tensor
special_sinc Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_sinc_t) Tensor
_self

special_round ::
  -- | self
  Tensor ->
  -- | decimals
  Int ->
  Tensor
special_round :: Tensor -> Int -> Tensor
special_round Tensor
_self Int
_decimals = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.special_round_tl) Tensor
_self Int
_decimals

special_log1p ::
  -- | self
  Tensor ->
  Tensor
special_log1p :: Tensor -> Tensor
special_log1p Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_log1p_t) Tensor
_self

special_log_softmax ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | dtype
  DType ->
  Tensor
special_log_softmax :: Tensor -> Int -> DType -> Tensor
special_log_softmax Tensor
_self Int
_dim DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> Int64 -> ScalarType -> IO (ForeignPtr Tensor)
ATen.special_log_softmax_tls) Tensor
_self Int
_dim DType
_dtype

special_gammainc ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
special_gammainc :: Tensor -> Tensor -> Tensor
special_gammainc Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_gammainc_tt) Tensor
_self Tensor
_other

special_gammaincc ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
special_gammaincc :: Tensor -> Tensor -> Tensor
special_gammaincc Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.special_gammaincc_tt) Tensor
_self Tensor
_other

special_multigammaln ::
  -- | self
  Tensor ->
  -- | p
  Int ->
  Tensor
special_multigammaln :: Tensor -> Int -> Tensor
special_multigammaln Tensor
_self Int
_p = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.special_multigammaln_tl) Tensor
_self Int
_p

special_softmax ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  -- | dtype
  DType ->
  Tensor
special_softmax :: Tensor -> Int -> DType -> Tensor
special_softmax Tensor
_self Int
_dim DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> Int64 -> ScalarType -> IO (ForeignPtr Tensor)
ATen.special_softmax_tls) Tensor
_self Int
_dim DType
_dtype

fft_fft ::
  -- | self
  Tensor ->
  -- | n
  Int ->
  -- | dim
  Int ->
  -- | norm
  String ->
  Tensor
fft_fft :: Tensor -> Int -> Int -> String -> Tensor
fft_fft Tensor
_self Int
_n Int
_dim String
_norm = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64 -> Int64 -> ForeignPtr StdString -> IO (ForeignPtr Tensor)
ATen.fft_fft_tlls) Tensor
_self Int
_n Int
_dim String
_norm

fft_ifft ::
  -- | self
  Tensor ->
  -- | n
  Int ->
  -- | dim
  Int ->
  -- | norm
  String ->
  Tensor
fft_ifft :: Tensor -> Int -> Int -> String -> Tensor
fft_ifft Tensor
_self Int
_n Int
_dim String
_norm = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64 -> Int64 -> ForeignPtr StdString -> IO (ForeignPtr Tensor)
ATen.fft_ifft_tlls) Tensor
_self Int
_n Int
_dim String
_norm

fft_rfft ::
  -- | self
  Tensor ->
  -- | n
  Int ->
  -- | dim
  Int ->
  -- | norm
  String ->
  Tensor
fft_rfft :: Tensor -> Int -> Int -> String -> Tensor
fft_rfft Tensor
_self Int
_n Int
_dim String
_norm = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64 -> Int64 -> ForeignPtr StdString -> IO (ForeignPtr Tensor)
ATen.fft_rfft_tlls) Tensor
_self Int
_n Int
_dim String
_norm

fft_irfft ::
  -- | self
  Tensor ->
  -- | n
  Int ->
  -- | dim
  Int ->
  -- | norm
  String ->
  Tensor
fft_irfft :: Tensor -> Int -> Int -> String -> Tensor
fft_irfft Tensor
_self Int
_n Int
_dim String
_norm = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64 -> Int64 -> ForeignPtr StdString -> IO (ForeignPtr Tensor)
ATen.fft_irfft_tlls) Tensor
_self Int
_n Int
_dim String
_norm

fft_hfft ::
  -- | self
  Tensor ->
  -- | n
  Int ->
  -- | dim
  Int ->
  -- | norm
  String ->
  Tensor
fft_hfft :: Tensor -> Int -> Int -> String -> Tensor
fft_hfft Tensor
_self Int
_n Int
_dim String
_norm = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64 -> Int64 -> ForeignPtr StdString -> IO (ForeignPtr Tensor)
ATen.fft_hfft_tlls) Tensor
_self Int
_n Int
_dim String
_norm

fft_ihfft ::
  -- | self
  Tensor ->
  -- | n
  Int ->
  -- | dim
  Int ->
  -- | norm
  String ->
  Tensor
fft_ihfft :: Tensor -> Int -> Int -> String -> Tensor
fft_ihfft Tensor
_self Int
_n Int
_dim String
_norm = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> Int64 -> Int64 -> ForeignPtr StdString -> IO (ForeignPtr Tensor)
ATen.fft_ihfft_tlls) Tensor
_self Int
_n Int
_dim String
_norm

fft_fft2 ::
  -- | self
  Tensor ->
  -- | s
  Int ->
  -- | dim
  Int ->
  -- | norm
  String ->
  Tensor
fft_fft2 :: Tensor -> Int -> Int -> String -> Tensor
fft_fft2 Tensor
_self Int
_s Int
_dim String
_norm = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr StdString
-> IO (ForeignPtr Tensor)
ATen.fft_fft2_tlls) Tensor
_self Int
_s Int
_dim String
_norm

fft_ifft2 ::
  -- | self
  Tensor ->
  -- | s
  Int ->
  -- | dim
  Int ->
  -- | norm
  String ->
  Tensor
fft_ifft2 :: Tensor -> Int -> Int -> String -> Tensor
fft_ifft2 Tensor
_self Int
_s Int
_dim String
_norm = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr StdString
-> IO (ForeignPtr Tensor)
ATen.fft_ifft2_tlls) Tensor
_self Int
_s Int
_dim String
_norm

fft_rfft2 ::
  -- | self
  Tensor ->
  -- | s
  Int ->
  -- | dim
  Int ->
  -- | norm
  String ->
  Tensor
fft_rfft2 :: Tensor -> Int -> Int -> String -> Tensor
fft_rfft2 Tensor
_self Int
_s Int
_dim String
_norm = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr StdString
-> IO (ForeignPtr Tensor)
ATen.fft_rfft2_tlls) Tensor
_self Int
_s Int
_dim String
_norm

fft_irfft2 ::
  -- | self
  Tensor ->
  -- | s
  Int ->
  -- | dim
  Int ->
  -- | norm
  String ->
  Tensor
fft_irfft2 :: Tensor -> Int -> Int -> String -> Tensor
fft_irfft2 Tensor
_self Int
_s Int
_dim String
_norm = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr StdString
-> IO (ForeignPtr Tensor)
ATen.fft_irfft2_tlls) Tensor
_self Int
_s Int
_dim String
_norm

fft_hfft2 ::
  -- | self
  Tensor ->
  -- | s
  Int ->
  -- | dim
  Int ->
  -- | norm
  String ->
  Tensor
fft_hfft2 :: Tensor -> Int -> Int -> String -> Tensor
fft_hfft2 Tensor
_self Int
_s Int
_dim String
_norm = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr StdString
-> IO (ForeignPtr Tensor)
ATen.fft_hfft2_tlls) Tensor
_self Int
_s Int
_dim String
_norm

fft_ihfft2 ::
  -- | self
  Tensor ->
  -- | s
  Int ->
  -- | dim
  Int ->
  -- | norm
  String ->
  Tensor
fft_ihfft2 :: Tensor -> Int -> Int -> String -> Tensor
fft_ihfft2 Tensor
_self Int
_s Int
_dim String
_norm = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr StdString
-> IO (ForeignPtr Tensor)
ATen.fft_ihfft2_tlls) Tensor
_self Int
_s Int
_dim String
_norm

fft_fftn ::
  -- | self
  Tensor ->
  -- | s
  Int ->
  -- | dim
  Int ->
  -- | norm
  String ->
  Tensor
fft_fftn :: Tensor -> Int -> Int -> String -> Tensor
fft_fftn Tensor
_self Int
_s Int
_dim String
_norm = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr StdString
-> IO (ForeignPtr Tensor)
ATen.fft_fftn_tlls) Tensor
_self Int
_s Int
_dim String
_norm

fft_ifftn ::
  -- | self
  Tensor ->
  -- | s
  Int ->
  -- | dim
  Int ->
  -- | norm
  String ->
  Tensor
fft_ifftn :: Tensor -> Int -> Int -> String -> Tensor
fft_ifftn Tensor
_self Int
_s Int
_dim String
_norm = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr StdString
-> IO (ForeignPtr Tensor)
ATen.fft_ifftn_tlls) Tensor
_self Int
_s Int
_dim String
_norm

fft_rfftn ::
  -- | self
  Tensor ->
  -- | s
  Int ->
  -- | dim
  Int ->
  -- | norm
  String ->
  Tensor
fft_rfftn :: Tensor -> Int -> Int -> String -> Tensor
fft_rfftn Tensor
_self Int
_s Int
_dim String
_norm = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr StdString
-> IO (ForeignPtr Tensor)
ATen.fft_rfftn_tlls) Tensor
_self Int
_s Int
_dim String
_norm

fft_irfftn ::
  -- | self
  Tensor ->
  -- | s
  Int ->
  -- | dim
  Int ->
  -- | norm
  String ->
  Tensor
fft_irfftn :: Tensor -> Int -> Int -> String -> Tensor
fft_irfftn Tensor
_self Int
_s Int
_dim String
_norm = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr StdString
-> IO (ForeignPtr Tensor)
ATen.fft_irfftn_tlls) Tensor
_self Int
_s Int
_dim String
_norm

fft_hfftn ::
  -- | self
  Tensor ->
  -- | s
  Int ->
  -- | dim
  Int ->
  -- | norm
  String ->
  Tensor
fft_hfftn :: Tensor -> Int -> Int -> String -> Tensor
fft_hfftn Tensor
_self Int
_s Int
_dim String
_norm = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr StdString
-> IO (ForeignPtr Tensor)
ATen.fft_hfftn_tlls) Tensor
_self Int
_s Int
_dim String
_norm

fft_ihfftn ::
  -- | self
  Tensor ->
  -- | s
  Int ->
  -- | dim
  Int ->
  -- | norm
  String ->
  Tensor
fft_ihfftn :: Tensor -> Int -> Int -> String -> Tensor
fft_ihfftn Tensor
_self Int
_s Int
_dim String
_norm = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr IntArray
-> ForeignPtr IntArray
-> ForeignPtr StdString
-> IO (ForeignPtr Tensor)
ATen.fft_ihfftn_tlls) Tensor
_self Int
_s Int
_dim String
_norm

fft_fftshift ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  Tensor
fft_fftshift :: Tensor -> Int -> Tensor
fft_fftshift Tensor
_self Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr IntArray -> IO (ForeignPtr Tensor)
ATen.fft_fftshift_tl) Tensor
_self Int
_dim

fft_ifftshift ::
  -- | self
  Tensor ->
  -- | dim
  Int ->
  Tensor
fft_ifftshift :: Tensor -> Int -> Tensor
fft_ifftshift Tensor
_self Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr IntArray -> IO (ForeignPtr Tensor)
ATen.fft_ifftshift_tl) Tensor
_self Int
_dim

linalg_cholesky_ex ::
  -- | self
  Tensor ->
  -- | upper
  Bool ->
  -- | check_errors
  Bool ->
  (Tensor, Tensor)
linalg_cholesky_ex :: Tensor -> Bool -> Bool -> (Tensor, Tensor)
linalg_cholesky_ex Tensor
_self Bool
_upper Bool
_check_errors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> CBool -> CBool -> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.linalg_cholesky_ex_tbb) Tensor
_self Bool
_upper Bool
_check_errors

linalg_cholesky ::
  -- | self
  Tensor ->
  -- | upper
  Bool ->
  Tensor
linalg_cholesky :: Tensor -> Bool -> Tensor
linalg_cholesky Tensor
_self Bool
_upper = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> CBool -> IO (ForeignPtr Tensor)
ATen.linalg_cholesky_tb) Tensor
_self Bool
_upper

linalg_cross ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  -- | dim
  Int ->
  Tensor
linalg_cross :: Tensor -> Tensor -> Int -> Tensor
linalg_cross Tensor
_self Tensor
_other Int
_dim = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.linalg_cross_ttl) Tensor
_self Tensor
_other Int
_dim

linalg_lu_factor ::
  -- | A
  Tensor ->
  -- | pivot
  Bool ->
  (Tensor, Tensor)
linalg_lu_factor :: Tensor -> Bool -> (Tensor, Tensor)
linalg_lu_factor Tensor
_A Bool
_pivot = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor
-> CBool -> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.linalg_lu_factor_tb) Tensor
_A Bool
_pivot

linalg_lu_factor_ex ::
  -- | A
  Tensor ->
  -- | pivot
  Bool ->
  -- | check_errors
  Bool ->
  (Tensor, Tensor, Tensor)
linalg_lu_factor_ex :: Tensor -> Bool -> Bool -> (Tensor, Tensor, Tensor)
linalg_lu_factor_ex Tensor
_A Bool
_pivot Bool
_check_errors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> CBool
-> CBool
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor, Tensor)))
ATen.linalg_lu_factor_ex_tbb) Tensor
_A Bool
_pivot Bool
_check_errors

linalg_det ::
  -- | self
  Tensor ->
  Tensor
linalg_det :: Tensor -> Tensor
linalg_det Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.linalg_det_t) Tensor
_self

det ::
  -- | self
  Tensor ->
  Tensor
det :: Tensor -> Tensor
det Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.det_t) Tensor
_self

linalg_lstsq ::
  -- | self
  Tensor ->
  -- | b
  Tensor ->
  -- | rcond
  Double ->
  -- | driver
  String ->
  (Tensor, Tensor, Tensor, Tensor)
linalg_lstsq :: Tensor
-> Tensor -> Double -> String -> (Tensor, Tensor, Tensor, Tensor)
linalg_lstsq Tensor
_self Tensor
_b Double
_rcond String
_driver = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Tensor
-> CDouble
-> ForeignPtr StdString
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor, Tensor, Tensor)))
ATen.linalg_lstsq_ttds) Tensor
_self Tensor
_b Double
_rcond String
_driver

linalg_matmul ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
linalg_matmul :: Tensor -> Tensor -> Tensor
linalg_matmul Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.linalg_matmul_tt) Tensor
_self Tensor
_other

linalg_matrix_exp ::
  -- | self
  Tensor ->
  Tensor
linalg_matrix_exp :: Tensor -> Tensor
linalg_matrix_exp Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.linalg_matrix_exp_t) Tensor
_self

linalg_slogdet ::
  -- | self
  Tensor ->
  (Tensor, Tensor)
linalg_slogdet :: Tensor -> (Tensor, Tensor)
linalg_slogdet Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.linalg_slogdet_t) Tensor
_self

linalg_eig ::
  -- | self
  Tensor ->
  (Tensor, Tensor)
linalg_eig :: Tensor -> (Tensor, Tensor)
linalg_eig Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.linalg_eig_t) Tensor
_self

linalg_eigvals ::
  -- | self
  Tensor ->
  Tensor
linalg_eigvals :: Tensor -> Tensor
linalg_eigvals Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.linalg_eigvals_t) Tensor
_self

linalg_eigh ::
  -- | self
  Tensor ->
  -- | UPLO
  String ->
  (Tensor, Tensor)
linalg_eigh :: Tensor -> String -> (Tensor, Tensor)
linalg_eigh Tensor
_self String
_UPLO = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor
-> ForeignPtr StdString
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.linalg_eigh_ts) Tensor
_self String
_UPLO

linalg_eigvalsh ::
  -- | self
  Tensor ->
  -- | UPLO
  String ->
  Tensor
linalg_eigvalsh :: Tensor -> String -> Tensor
linalg_eigvalsh Tensor
_self String
_UPLO = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr StdString -> IO (ForeignPtr Tensor)
ATen.linalg_eigvalsh_ts) Tensor
_self String
_UPLO

linalg_householder_product ::
  -- | input
  Tensor ->
  -- | tau
  Tensor ->
  Tensor
linalg_householder_product :: Tensor -> Tensor -> Tensor
linalg_householder_product Tensor
_input Tensor
_tau = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.linalg_householder_product_tt) Tensor
_input Tensor
_tau

linalg_inv_ex ::
  -- | self
  Tensor ->
  -- | check_errors
  Bool ->
  (Tensor, Tensor)
linalg_inv_ex :: Tensor -> Bool -> (Tensor, Tensor)
linalg_inv_ex Tensor
_self Bool
_check_errors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor
-> CBool -> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.linalg_inv_ex_tb) Tensor
_self Bool
_check_errors

linalg_inv ::
  -- | self
  Tensor ->
  Tensor
linalg_inv :: Tensor -> Tensor
linalg_inv Tensor
_self = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.linalg_inv_t) Tensor
_self

inner ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  Tensor
inner :: Tensor -> Tensor -> Tensor
inner Tensor
_self Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.inner_tt) Tensor
_self Tensor
_other

outer ::
  -- | self
  Tensor ->
  -- | vec2
  Tensor ->
  Tensor
outer :: Tensor -> Tensor -> Tensor
outer Tensor
_self Tensor
_vec2 = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.outer_tt) Tensor
_self Tensor
_vec2

ger ::
  -- | self
  Tensor ->
  -- | vec2
  Tensor ->
  Tensor
ger :: Tensor -> Tensor -> Tensor
ger Tensor
_self Tensor
_vec2 = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.ger_tt) Tensor
_self Tensor
_vec2

linalg_norm_tslbs ::
  -- | self
  Tensor ->
  -- | ord
  Float ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  -- | dtype
  DType ->
  Tensor
linalg_norm_tslbs :: Tensor -> Float -> Int -> Bool -> DType -> Tensor
linalg_norm_tslbs Tensor
_self Float
_ord Int
_dim Bool
_keepdim DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr IntArray
-> CBool
-> ScalarType
-> IO (ForeignPtr Tensor)
ATen.linalg_norm_tslbs) Tensor
_self Float
_ord Int
_dim Bool
_keepdim DType
_dtype

-- linalg_norm_tslbs
--   :: Tensor -- ^ self
--   -> String -- ^ ord
--   -> Int -- ^ dim
--   -> Bool -- ^ keepdim
--   -> DType -- ^ dtype
--   -> Tensor
-- linalg_norm_tslbs _self _ord _dim _keepdim _dtype = unsafePerformIO $ (cast5 ATen.linalg_norm_tslbs) _self _ord _dim _keepdim _dtype

linalg_vector_norm ::
  -- | self
  Tensor ->
  -- | ord
  Float ->
  -- | dim
  Int ->
  -- | keepdim
  Bool ->
  -- | dtype
  DType ->
  Tensor
linalg_vector_norm :: Tensor -> Float -> Int -> Bool -> DType -> Tensor
linalg_vector_norm Tensor
_self Float
_ord Int
_dim Bool
_keepdim DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr IntArray
-> CBool
-> ScalarType
-> IO (ForeignPtr Tensor)
ATen.linalg_vector_norm_tslbs) Tensor
_self Float
_ord Int
_dim Bool
_keepdim DType
_dtype

linalg_matrix_norm_tslbs ::
  -- | self
  Tensor ->
  -- | ord
  Float ->
  -- | dim
  [Int] ->
  -- | keepdim
  Bool ->
  -- | dtype
  DType ->
  Tensor
linalg_matrix_norm_tslbs :: Tensor -> Float -> [Int] -> Bool -> DType -> Tensor
linalg_matrix_norm_tslbs Tensor
_self Float
_ord [Int]
_dim Bool
_keepdim DType
_dtype = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> IO y
cast5 ForeignPtr Tensor
-> ForeignPtr Scalar
-> ForeignPtr IntArray
-> CBool
-> ScalarType
-> IO (ForeignPtr Tensor)
ATen.linalg_matrix_norm_tslbs) Tensor
_self Float
_ord [Int]
_dim Bool
_keepdim DType
_dtype

-- linalg_matrix_norm_tslbs
--   :: Tensor -- ^ self
--   -> String -- ^ ord
--   -> [Int] -- ^ dim
--   -> Bool -- ^ keepdim
--   -> DType -- ^ dtype
--   -> Tensor
-- linalg_matrix_norm_tslbs _self _ord _dim _keepdim _dtype = unsafePerformIO $ (cast5 ATen.linalg_matrix_norm_tslbs) _self _ord _dim _keepdim _dtype

linalg_svd ::
  -- | A
  Tensor ->
  -- | full_matrices
  Bool ->
  (Tensor, Tensor, Tensor)
linalg_svd :: Tensor -> Bool -> (Tensor, Tensor, Tensor)
linalg_svd Tensor
_A Bool
_full_matrices = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor
-> CBool -> IO (ForeignPtr (StdTuple '(Tensor, Tensor, Tensor)))
ATen.linalg_svd_tb) Tensor
_A Bool
_full_matrices

linalg_svdvals ::
  -- | A
  Tensor ->
  Tensor
linalg_svdvals :: Tensor -> Tensor
linalg_svdvals Tensor
_A = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.linalg_svdvals_t) Tensor
_A

linalg_cond_ts ::
  -- | self
  Tensor ->
  -- | p
  Float ->
  Tensor
linalg_cond_ts :: Tensor -> Float -> Tensor
linalg_cond_ts Tensor
_self Float
_p = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Scalar -> IO (ForeignPtr Tensor)
ATen.linalg_cond_ts) Tensor
_self Float
_p

-- linalg_cond_ts
--   :: Tensor -- ^ self
--   -> String -- ^ p
--   -> Tensor
-- linalg_cond_ts _self _p = unsafePerformIO $ (cast2 ATen.linalg_cond_ts) _self _p

linalg_pinv_tttb ::
  -- | self
  Tensor ->
  -- | atol
  Tensor ->
  -- | rtol
  Tensor ->
  -- | hermitian
  Bool ->
  Tensor
linalg_pinv_tttb :: Tensor -> Tensor -> Tensor -> Bool -> Tensor
linalg_pinv_tttb Tensor
_self Tensor
_atol Tensor
_rtol Bool
_hermitian = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> CBool
-> IO (ForeignPtr Tensor)
ATen.linalg_pinv_tttb) Tensor
_self Tensor
_atol Tensor
_rtol Bool
_hermitian

linalg_pinv_tddb ::
  -- | self
  Tensor ->
  -- | atol
  Double ->
  -- | rtol
  Double ->
  -- | hermitian
  Bool ->
  Tensor
linalg_pinv_tddb :: Tensor -> Double -> Double -> Bool -> Tensor
linalg_pinv_tddb Tensor
_self Double
_atol Double
_rtol Bool
_hermitian = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> CDouble -> CDouble -> CBool -> IO (ForeignPtr Tensor)
ATen.linalg_pinv_tddb) Tensor
_self Double
_atol Double
_rtol Bool
_hermitian

linalg_pinv_tdb ::
  -- | self
  Tensor ->
  -- | rcond
  Double ->
  -- | hermitian
  Bool ->
  Tensor
linalg_pinv_tdb :: Tensor -> Double -> Bool -> Tensor
linalg_pinv_tdb Tensor
_self Double
_rcond Bool
_hermitian = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> CDouble -> CBool -> IO (ForeignPtr Tensor)
ATen.linalg_pinv_tdb) Tensor
_self Double
_rcond Bool
_hermitian

linalg_pinv_ttb ::
  -- | self
  Tensor ->
  -- | rcond
  Tensor ->
  -- | hermitian
  Bool ->
  Tensor
linalg_pinv_ttb :: Tensor -> Tensor -> Bool -> Tensor
linalg_pinv_ttb Tensor
_self Tensor
_rcond Bool
_hermitian = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> CBool -> IO (ForeignPtr Tensor)
ATen.linalg_pinv_ttb) Tensor
_self Tensor
_rcond Bool
_hermitian

linalg_solve ::
  -- | input
  Tensor ->
  -- | other
  Tensor ->
  Tensor
linalg_solve :: Tensor -> Tensor -> Tensor
linalg_solve Tensor
_input Tensor
_other = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> ForeignPtr Tensor -> IO (ForeignPtr Tensor)
ATen.linalg_solve_tt) Tensor
_input Tensor
_other

linalg_tensorinv ::
  -- | self
  Tensor ->
  -- | ind
  Int ->
  Tensor
linalg_tensorinv :: Tensor -> Int -> Tensor
linalg_tensorinv Tensor
_self Int
_ind = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.linalg_tensorinv_tl) Tensor
_self Int
_ind

linalg_tensorsolve ::
  -- | self
  Tensor ->
  -- | other
  Tensor ->
  -- | dims
  [Int] ->
  Tensor
linalg_tensorsolve :: Tensor -> Tensor -> [Int] -> Tensor
linalg_tensorsolve Tensor
_self Tensor
_other [Int]
_dims = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr IntArray
-> IO (ForeignPtr Tensor)
ATen.linalg_tensorsolve_ttl) Tensor
_self Tensor
_other [Int]
_dims

linalg_qr ::
  -- | self
  Tensor ->
  -- | mode
  String ->
  (Tensor, Tensor)
linalg_qr :: Tensor -> String -> (Tensor, Tensor)
linalg_qr Tensor
_self String
_mode = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor
-> ForeignPtr StdString
-> IO (ForeignPtr (StdTuple '(Tensor, Tensor)))
ATen.linalg_qr_ts) Tensor
_self String
_mode

linalg_matrix_power ::
  -- | self
  Tensor ->
  -- | n
  Int ->
  Tensor
linalg_matrix_power :: Tensor -> Int -> Tensor
linalg_matrix_power Tensor
_self Int
_n = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor -> Int64 -> IO (ForeignPtr Tensor)
ATen.linalg_matrix_power_tl) Tensor
_self Int
_n

linalg_matrix_rank_tttb ::
  -- | input
  Tensor ->
  -- | atol
  Tensor ->
  -- | rtol
  Tensor ->
  -- | hermitian
  Bool ->
  Tensor
linalg_matrix_rank_tttb :: Tensor -> Tensor -> Tensor -> Bool -> Tensor
linalg_matrix_rank_tttb Tensor
_input Tensor
_atol Tensor
_rtol Bool
_hermitian = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> CBool
-> IO (ForeignPtr Tensor)
ATen.linalg_matrix_rank_tttb) Tensor
_input Tensor
_atol Tensor
_rtol Bool
_hermitian

linalg_matrix_rank_tddb ::
  -- | self
  Tensor ->
  -- | atol
  Double ->
  -- | rtol
  Double ->
  -- | hermitian
  Bool ->
  Tensor
linalg_matrix_rank_tddb :: Tensor -> Double -> Double -> Bool -> Tensor
linalg_matrix_rank_tddb Tensor
_self Double
_atol Double
_rtol Bool
_hermitian = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> IO cy) -> a -> x1 -> x2 -> x3 -> IO y
cast4 ForeignPtr Tensor
-> CDouble -> CDouble -> CBool -> IO (ForeignPtr Tensor)
ATen.linalg_matrix_rank_tddb) Tensor
_self Double
_atol Double
_rtol Bool
_hermitian

linalg_matrix_rank_tdb ::
  -- | self
  Tensor ->
  -- | tol
  Double ->
  -- | hermitian
  Bool ->
  Tensor
linalg_matrix_rank_tdb :: Tensor -> Double -> Bool -> Tensor
linalg_matrix_rank_tdb Tensor
_self Double
_tol Bool
_hermitian = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor -> CDouble -> CBool -> IO (ForeignPtr Tensor)
ATen.linalg_matrix_rank_tdb) Tensor
_self Double
_tol Bool
_hermitian

linalg_matrix_rank_ttb ::
  -- | input
  Tensor ->
  -- | tol
  Tensor ->
  -- | hermitian
  Bool ->
  Tensor
linalg_matrix_rank_ttb :: Tensor -> Tensor -> Bool -> Tensor
linalg_matrix_rank_ttb Tensor
_input Tensor
_tol Bool
_hermitian = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr Tensor
-> ForeignPtr Tensor -> CBool -> IO (ForeignPtr Tensor)
ATen.linalg_matrix_rank_ttb) Tensor
_input Tensor
_tol Bool
_hermitian

linalg_multi_dot ::
  -- | tensors
  [Tensor] ->
  Tensor
linalg_multi_dot :: [Tensor] -> Tensor
linalg_multi_dot [Tensor]
_tensors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr TensorList -> IO (ForeignPtr Tensor)
ATen.linalg_multi_dot_l) [Tensor]
_tensors

segment_reduce ::
  -- | data
  Tensor ->
  -- | reduce
  String ->
  -- | lengths
  Tensor ->
  -- | indices
  Tensor ->
  -- | axis
  Int ->
  -- | unsafe
  Bool ->
  -- | initial
  Float ->
  Tensor
segment_reduce :: Tensor
-> String -> Tensor -> Tensor -> Int -> Bool -> Float -> Tensor
segment_reduce Tensor
_data String
_reduce Tensor
_lengths Tensor
_indices Int
_axis Bool
_unsafe Float
_initial = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 x3 cx3 x4 cx4 x5 cx5 x6 cx6 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable x3 cx3,
 Castable x4 cx4, Castable x5 cx5, Castable x6 cx6,
 Castable y cy) =>
(ca -> cx1 -> cx2 -> cx3 -> cx4 -> cx5 -> cx6 -> IO cy)
-> a -> x1 -> x2 -> x3 -> x4 -> x5 -> x6 -> IO y
cast7 ForeignPtr Tensor
-> ForeignPtr StdString
-> ForeignPtr Tensor
-> ForeignPtr Tensor
-> Int64
-> CBool
-> ForeignPtr Scalar
-> IO (ForeignPtr Tensor)
ATen.segment_reduce_tsttlbs) Tensor
_data String
_reduce Tensor
_lengths Tensor
_indices Int
_axis Bool
_unsafe Float
_initial

pad_sequence ::
  -- | sequences
  [Tensor] ->
  -- | batch_first
  Bool ->
  -- | padding_value
  Double ->
  Tensor
pad_sequence :: [Tensor] -> Bool -> Double -> Tensor
pad_sequence [Tensor]
_sequences Bool
_batch_first Double
_padding_value = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 x2 cx2 y cy.
(Castable a ca, Castable x1 cx1, Castable x2 cx2, Castable y cy) =>
(ca -> cx1 -> cx2 -> IO cy) -> a -> x1 -> x2 -> IO y
cast3 ForeignPtr TensorList -> CBool -> CDouble -> IO (ForeignPtr Tensor)
ATen.pad_sequence_lbd) [Tensor]
_sequences Bool
_batch_first Double
_padding_value

flatten_dense_tensors ::
  -- | tensors
  [Tensor] ->
  Tensor
flatten_dense_tensors :: [Tensor] -> Tensor
flatten_dense_tensors [Tensor]
_tensors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca y cy.
(Castable a ca, Castable y cy) =>
(ca -> IO cy) -> a -> IO y
cast1 ForeignPtr TensorList -> IO (ForeignPtr Tensor)
ATen.flatten_dense_tensors_l) [Tensor]
_tensors

unflatten_dense_tensors ::
  -- | flat
  Tensor ->
  -- | tensors
  [Tensor] ->
  [Tensor]
unflatten_dense_tensors :: Tensor -> [Tensor] -> [Tensor]
unflatten_dense_tensors Tensor
_flat [Tensor]
_tensors = forall a. IO a -> a
unsafePerformIO forall a b. (a -> b) -> a -> b
$ (forall a ca x1 cx1 y cy.
(Castable a ca, Castable x1 cx1, Castable y cy) =>
(ca -> cx1 -> IO cy) -> a -> x1 -> IO y
cast2 ForeignPtr Tensor
-> ForeignPtr TensorList -> IO (ForeignPtr TensorList)
ATen.unflatten_dense_tensors_tl) Tensor
_flat [Tensor]
_tensors