ops

Structs

Functions

add


add(x: Array, y: Array) -> Array

Args

x: Array
y: Array

Returns

Array

mul


mul(x: Array, y: Array) -> Array

Args

x: Array
y: Array

Returns

Array

sub


sub(x: Array, y: Array) -> Array

Args

x: Array
y: Array

Returns

Array

div


div(x: Array, y: Array) -> Array

Args

x: Array
y: Array

Returns

Array

matmul


matmul(x: Array, y: Array) -> Array

Args

x: Array
y: Array

Returns

Array

gt


gt(x: Array, y: Array) -> Array

Args

x: Array
y: Array

Returns

Array

pow


pow(x: Array, y: Array) -> Array

Args

x: Array
y: Array

Returns

Array

sum


sum(x: Array, axis: List[Int] = List(), keep_dim: Bool = False, act_on_batch_dims: Bool = False) -> Array

Args

x: Array
axis: List[Int] (default: List())
keep_dim: Bool (default: False)
act_on_batch_dims: Bool (default: False)

Returns

Array

sin


sin(x: Array) -> Array

Args

x: Array

Returns

Array

cast


cast(x: Array, dtype: DType) -> Array

Args

x: Array
dtype: DType

Returns

Array

negate


negate(x: Array) -> Array

Args

x: Array

Returns

Array

cos


cos(x: Array) -> Array

Args

x: Array

Returns

Array

relu


relu(x: Array) -> Array

Args

x: Array

Returns

Array

log


log(x: Array) -> Array

Args

x: Array

Returns

Array

gt_zero


gt_zero(x: Array) -> Array

Args

x: Array

Returns

Array

incr_batch_dim_ctr


incr_batch_dim_ctr(x: Array) -> Array

Args

x: Array

Returns

Array

decr_batch_dim_ctr


decr_batch_dim_ctr(x: Array) -> Array

Args

x: Array

Returns

Array

permute


permute(arg: Array, perm: List[Int]) -> Array

Args

arg: Array
perm: List[Int]

Returns

Array

transpose


transpose(arg: Array, x: Int, y: Int) -> Array

Args

arg: Array
x: Int
y: Int

Returns

Array

reshape


reshape(x: Array, shape: Variant[List[Int], Tuple[Int], Tuple[Int, Int], Tuple[Int, Int, Int], Tuple[Int, Int, Int, Int], Tuple[Int, Int, Int, Int, Int], Tuple[Int, Int, Int, Int, Int, Int], Tuple[Int, Int, Int, Int, Int, Int, Int], Tuple[Int, Int, Int, Int, Int, Int, Int, Int]]) -> Array

Args

x: Array
shape: Variant[List[Int], Tuple[Int], Tuple[Int, Int], Tuple[Int, Int, Int], Tuple[Int, Int, Int, Int], Tuple[Int, Int, Int, Int, Int], Tuple[Int, Int, Int, Int, Int, Int], Tuple[Int, Int, Int, Int, Int, Int, Int], Tuple[Int, Int, Int, Int, Int, Int, Int, Int]]

Returns

Array

flatten


flatten(x: Array) -> Array

Args

x: Array

Returns

Array

broadcast_to


broadcast_to(x: Array, shape: Variant[List[Int], Tuple[Int], Tuple[Int, Int], Tuple[Int, Int, Int], Tuple[Int, Int, Int, Int], Tuple[Int, Int, Int, Int, Int], Tuple[Int, Int, Int, Int, Int, Int], Tuple[Int, Int, Int, Int, Int, Int, Int], Tuple[Int, Int, Int, Int, Int, Int, Int, Int]], act_on_batch_dims: Bool = False) -> Array

Args

x: Array
shape: Variant[List[Int], Tuple[Int], Tuple[Int, Int], Tuple[Int, Int, Int], Tuple[Int, Int, Int, Int], Tuple[Int, Int, Int, Int, Int], Tuple[Int, Int, Int, Int, Int, Int], Tuple[Int, Int, Int, Int, Int, Int, Int], Tuple[Int, Int, Int, Int, Int, Int, Int, Int]]
act_on_batch_dims: Bool (default: False)

Returns

Array

stack


stack(args: List[Array], axis: Int = 0) -> Array

Args

args: List[Array]
axis: Int (default: 0)

Returns

Array

array_slice


array_slice(arg: Array, slices: List[Slice]) -> Array

Args

arg: Array
slices: List[Slice]

Returns

Array

concat


concat(args: List[Array], axis: Int = 0) -> Array

Args

args: List[Array]
axis: Int (default: 0)

Returns

Array

split


split(arg: Array, sizes: List[Int], axis: Int) -> List[Array]

Args

arg: Array
sizes: List[Int]
axis: Int

Returns

List[Array]

squeeze


squeeze(arg: Array, axis: List[Int], act_on_batch_dims: Bool = False) -> Array

Args

arg: Array
axis: List[Int]
act_on_batch_dims: Bool (default: False)

Returns

Array

unsqueeze


unsqueeze(arg: Array, axes: List[Int], act_on_batch_dims: Bool = False) -> Array

Args

arg: Array
axes: List[Int]
act_on_batch_dims: Bool (default: False)

Returns

Array

backward


backward(array: Array, remat: Bool = False)

Args

array: Array
remat: Bool (default: False)