{-# 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 ::
[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 ::
Tensor ->
Double ->
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 ::
Tensor ->
Double ->
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 ::
Tensor ->
Double ->
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 ::
Tensor ->
Double ->
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 ::
Tensor ->
Double ->
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 ::
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 ::
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 ::
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 ::
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 ::
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 ::
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 ::
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 ::
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 ::
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 ::
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 ::
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 ::
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 ::
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 ::
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 ::
Tensor ->
Int ->
Int ->
Int ->
Bool ->
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 ::
Tensor ->
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 ::
Tensor ->
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 ::
Tensor ->
Tensor ->
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 ::
Tensor ->
Float ->
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 ::
Tensor ->
Tensor ->
Tensor ->
Float ->
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 ::
Tensor ->
Tensor ->
Tensor ->
Float ->
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 ::
Tensor ->
[Int] ->
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 ::
Tensor ->
Int ->
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 ::
Tensor ->
Dimname ->
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 ::
Tensor ->
Tensor ->
Double ->
Double ->
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 ::
Tensor ->
Int ->
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 ::
Tensor ->
Dimname ->
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 ::
Tensor ->
Int ->
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 ::
Tensor ->
Int ->
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 ::
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 ::
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 ::
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 ::
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 ::
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 ::
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 ::
Tensor ->
[Int] ->
[Int] ->
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 ::
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 ::
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 ::
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 ::
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 ::
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 ::
[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 ::
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 ::
[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 ::
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 ::
[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 ::
Tensor ->
Tensor ->
Tensor ->
Float ->
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 ::
Tensor ->
Tensor ->
Tensor ->
Tensor ->
Tensor ->
Bool ->
Double ->
Double ->
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 ::
Tensor ->
Tensor ->
Tensor ->
Tensor ->
Tensor ->
Double ->
Double ->
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 ::
Tensor ->
Tensor ->
Tensor ->
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 ::
Tensor ->
Tensor ->
Tensor ->
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 ::
Tensor ->
Tensor ->
Tensor ->
Tensor ->
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 ::
Tensor ->
Tensor ->
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 ::
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 ::
Tensor ->
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 ::
Tensor ->
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 ::
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 ::
Tensor ->
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 ::
Tensor ->
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 ::
Tensor ->
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 ::
Tensor ->
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 ::
[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 ::
Tensor ->
[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 ::
[Tensor] ->
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 ::
[Tensor] ->
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 ::
[Tensor] ->
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 ::
[Tensor] ->
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 ::
[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 ::
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 ::
[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 ::
Tensor ->
Int ->
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 ::
Tensor ->
Int ->
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 ::
Tensor ->
Int ->
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
clamp_tss ::
Tensor ->
Float ->
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 ::
Tensor ->
Tensor ->
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 ::
Tensor ->
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 ::
Tensor ->
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 ::
Tensor ->
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 ::
Tensor ->
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 ::
Tensor ->
Float ->
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 ::
Tensor ->
Tensor ->
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 ::
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 ::
Tensor ->
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 ::
Tensor ->
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 ::
Tensor ->
[Int] ->
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 ::
Tensor ->
Tensor ->
Tensor ->
[Int] ->
[Int] ->
[Int] ->
Bool ->
[Int] ->
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 ::
Tensor ->
Tensor ->
Tensor ->
[Int] ->
[Int] ->
[Int] ->
Bool ->
[Int] ->
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 ::
Tensor ->
Tensor ->
Tensor ->
[Int] ->
[Int] ->
[Int] ->
Bool ->
[Int] ->
Int ->
(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 ::
Tensor ->
Tensor ->
Tensor ->
Int ->
Int ->
Int ->
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 ::
Tensor ->
Tensor ->
Tensor ->
(Int, Int) ->
(Int, Int) ->
(Int, Int) ->
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 ::
Tensor ->
Tensor ->
Tensor ->
(Int, Int, Int) ->
(Int, Int, Int) ->
(Int, Int, Int) ->
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 ::
Tensor ->
Tensor ->
Tensor ->
Int ->
String ->
Int ->
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 ::
Tensor ->
Tensor ->
Tensor ->
(Int, Int) ->
String ->
(Int, Int) ->
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 ::
Tensor ->
Tensor ->
Tensor ->
(Int, Int, Int) ->
String ->
(Int, Int, Int) ->
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 ::
Tensor ->
Tensor ->
Tensor ->
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 ::
Tensor ->
Tensor ->
Tensor ->
Int ->
Int ->
Int ->
Int ->
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 ::
Tensor ->
Tensor ->
Tensor ->
(Int, Int) ->
(Int, Int) ->
(Int, Int) ->
Int ->
(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 ::
Tensor ->
Tensor ->
Tensor ->
(Int, Int, Int) ->
(Int, Int, Int) ->
(Int, Int, Int) ->
Int ->
(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 ::
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 ::
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 ::
Tensor ->
Tensor ->
Tensor ->
Double ->
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 ::
Tensor ->
[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
cov ::
Tensor ->
Int ->
Tensor ->
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 ::
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 ::
Tensor ->
Int ->
Int ->
Int ->
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 ::
Tensor ->
Tensor ->
Tensor ->
Tensor ->
Tensor ->
Bool ->
Double ->
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 ::
Tensor ->
Tensor ->
[Int] ->
[Int] ->
[Int] ->
Int ->
Bool ->
Bool ->
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 ::
Tensor ->
Tensor ->
[Int] ->
[Int] ->
[Int] ->
[Int] ->
Int ->
Bool ->
Bool ->
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 ::
Tensor ->
Tensor ->
Tensor ->
[Int] ->
[Int] ->
[Int] ->
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 ::
Tensor ->
Tensor ->
Tensor ->
Float ->
Tensor ->
[Int] ->
[Int] ->
[Int] ->
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 ::
Tensor ->
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 ::
Tensor ->
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 ::
Tensor ->
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 ::
Tensor ->
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 ::
Tensor ->
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 ::
Tensor ->
Int ->
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 ::
Tensor ->
Dimname ->
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 ::
Tensor ->
Int ->
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 ::
Tensor ->
Dimname ->
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 ::
Tensor ->
Tensor ->
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 ::
Tensor ->
Float ->
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' ::
Tensor ->
Tensor ->
[Int] ->
[Int] ->
Int ->
Int ->
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 ::
Tensor ->
Tensor ->
Tensor ->
Tensor ->
Int ->
Int ->
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 ::
Tensor ->
Int ->
Int ->
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 ::
Tensor ->
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 ::
Tensor ->
Int ->
Int ->
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 ::
Tensor ->
Int ->
Int ->
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 ::
Tensor ->
Dimname ->
Dimname ->
Dimname ->
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 ::
Tensor ->
Int ->
Int ->
Tensor ->
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 ::
Tensor ->
Float ->
Int ->
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_tll ::
Tensor ->
[Int] ->
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
div ::
Tensor ->
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 ::
Tensor ->
Tensor ->
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 ::
Tensor ->
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 ::
Tensor ->
Float ->
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 ::
Tensor ->
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 ::
Tensor ->
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 ::
Tensor ->
Tensor ->
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 ::
Tensor ->
Float ->
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 ::
Tensor ->
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 ::
Tensor ->
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 ::
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 ::
Tensor ->
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 ::
String ->
[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 ::
Tensor ->
Tensor ->
Int ->
Bool ->
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 ::
[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 ::
Tensor ->
Tensor ->
Tensor ->
Bool ->
Int ->
Bool ->
Tensor ->
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 ::
Tensor ->
Tensor ->
Tensor ->
Bool ->
Int ->
Bool ->
Tensor ->
Bool ->
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 ::
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 ::
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 ::
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 ::
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 ::
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 ::
Tensor ->
Int ->
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 ::
Tensor ->
Int ->
Int ->
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 ::
Tensor ->
Dimname ->
Dimname ->
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 ::
Tensor ->
[Dimname] ->
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 ::
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 ::
Tensor ->
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 ::
Tensor ->
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 ::
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 ::
Tensor ->
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 ::
Tensor ->
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 ::
Tensor ->
Tensor ->
Int ->
Int ->
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 ::
Tensor ->
Tensor ->
Int ->
Int ->
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 ::
Tensor ->
Tensor ->
Int ->
Int ->
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 ::
Tensor ->
Tensor ->
Double ->
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 ::
Tensor ->
Int ->
Tensor ->
Tensor ->
Double ->
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 ::
Tensor ->
Tensor ->
Tensor ->
Int ->
Int ->
Int ->
Int ->
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 ::
Tensor ->
[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 ::
Tensor ->
Int ->
Tensor ->
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 ::
Tensor ->
Dimname ->
Tensor ->
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 ::
Tensor ->
[Tensor] ->
Tensor ->
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 ::
Tensor ->
Tensor ->
Tensor ->
Tensor ->
Tensor ->
Bool ->
Double ->
Double ->
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 ::
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 ::
Tensor ->
Tensor ->
Double ->
Double ->
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 ::
Tensor ->
Tensor ->
Bool ->
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 ::
Tensor ->
Float ->
Bool ->
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 ::
Float ->
Tensor ->
Bool ->
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 ::
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 ::
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 ::
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 ::
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 ::
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 ::
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 ::
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 ::
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 ::
Tensor ->
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 ::
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 ::
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 ::
Tensor ->
Tensor ->
Int ->
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 ::
Tensor ->
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 ::
Tensor ->
Int ->
Int ->
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 ::
Tensor ->
Int ->
Dimname ->
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 ::
Tensor ->
[Int] ->
Tensor ->
Tensor ->
Double ->
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 ::
Tensor ->
[Int] ->
Tensor ->
Tensor ->
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 ::
Tensor ->
Double ->
Double ->
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 ::
Tensor ->
Tensor ->
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 ::
Tensor ->
Tensor ->
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 ::
[Int] ->
Tensor ->
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 ::
Tensor ->
Tensor ->
Tensor ->
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 ::
Tensor ->
Tensor ->
Tensor ->
Tensor ->
Float ->
Float ->
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 ::
Tensor ->
Tensor ->
Tensor ->
Tensor ->
Float ->
Float ->
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 ::
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 ::
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 ::
Tensor ->
Tensor ->
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 ::
Tensor ->
Tensor ->
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' ::
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 ::
Tensor ->
Int ->
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 ::
Tensor ->
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 ::
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 ::
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 ::
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 ::
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 ::
Tensor ->
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 ::
Tensor ->
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 ::
Tensor ->
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 ::
Float ->
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 ::
Tensor ->
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 ::
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 ::
Tensor ->
Int ->
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 ::
Tensor ->
Dimname ->
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 ::
Tensor ->
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 ::
Tensor ->
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 ::
Tensor ->
Int ->
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 ::
Tensor ->
[Dimname] ->
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 ::
Tensor ->
Tensor ->
Tensor ->
Double ->
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 ::
Tensor ->
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 ::
Tensor ->
Double ->
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' ::
Tensor ->
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 ::
Tensor ->
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 ::
Tensor ->
Tensor
matrix_exp :: Tensor -> Tensor
matrix_exp Tensor
_self = forall a. IO a -> a
unsafePerformIO