Tensor Class Reference


                    
                    
                    rawValue

Get the underlying tensor. If not available, create one.

Declaration

Swift

public var rawValue: NNC.Tensor<Element> { get }


                    
                    
                    subscript(_:)

Declaration

Swift

public subscript(indices: Int...) -> Element { get set }


                    
                    
                    subscript(_:streamContext:)

Declaration

Swift

public subscript(ranges: Range<Int>..., streamContext streamContext: StreamContext? = nil)
  -> DynamicGraph.Tensor<Element>


                    
                    
                    subscript(_:)

Declaration

Swift

@inlinable
public subscript(range: Range<Int>) -> DynamicGraph.Tensor<Element> { get set }


                    
                    
                    subscript(_:_:)

Declaration

Swift

@inlinable
public subscript(i0: Int, range: Range<Int>) -> DynamicGraph.Tensor<Element> { get set }


                    
                    
                    subscript(_:_:_:)

Declaration

Swift

@inlinable
public subscript(i0: Int, i1: Int, range: Range<Int>) -> DynamicGraph.Tensor<Element> { get set }


                    
                    
                    subscript(_:_:_:_:)

Declaration

Swift

@inlinable
public subscript(i0: Int, i1: Int, i2: Int, range: Range<Int>) -> DynamicGraph.Tensor<Element> { get set }


                    
                    
                    subscript(_:_:_:_:_:)

Declaration

Swift

@inlinable
public subscript(i0: Int, i1: Int, i2: Int, i3: Int, range: Range<Int>)
  -> DynamicGraph.Tensor<Element>


                    
                    
                    subscript(_:_:_:_:_:_:)

Declaration

Swift

@inlinable
public subscript(i0: Int, i1: Int, i2: Int, i3: Int, i4: Int, range: Range<Int>)
  -> DynamicGraph.Tensor<
    Element
  >


                    
                    
                    subscript(_:_:_:_:_:_:_:)

Declaration

Swift

@inlinable
public subscript(i0: Int, i1: Int, i2: Int, i3: Int, i4: Int, i5: Int, range: Range<Int>)
  -> DynamicGraph.Tensor<Element>


                    
                    
                    subscript(_:_:_:_:_:_:_:_:)

Declaration

Swift

@inlinable
public subscript(i0: Int, i1: Int, i2: Int, i3: Int, i4: Int, i5: Int, i6: Int, range: Range<Int>)
  -> DynamicGraph.Tensor<Element>


                    
                    
                    subscript(_:)

Declaration

Swift

@inlinable
public subscript(range: UnboundedRange) -> DynamicGraph.Tensor<Element> { get set }


                    
                    
                    subscript(_:_:)

Declaration

Swift

@inlinable
public subscript(i0: Int, range: UnboundedRange) -> DynamicGraph.Tensor<Element> { get set }


                    
                    
                    subscript(_:_:_:)

Declaration

Swift

@inlinable
public subscript(i0: Int, i1: Int, range: UnboundedRange) -> DynamicGraph.Tensor<Element> { get set }


                    
                    
                    subscript(_:_:_:_:)

Declaration

Swift

@inlinable
public subscript(i0: Int, i1: Int, i2: Int, range: UnboundedRange) -> DynamicGraph.Tensor<Element> { get set }


                    
                    
                    subscript(_:_:_:_:_:)

Declaration

Swift

@inlinable
public subscript(i0: Int, i1: Int, i2: Int, i3: Int, range: UnboundedRange)
  -> DynamicGraph.Tensor<Element>


                    
                    
                    subscript(_:_:_:_:_:_:)

Declaration

Swift

@inlinable
public subscript(i0: Int, i1: Int, i2: Int, i3: Int, i4: Int, range: UnboundedRange)
  -> DynamicGraph.Tensor<
    Element
  >


                    
                    
                    subscript(_:_:_:_:_:_:_:)

Declaration

Swift

@inlinable
public subscript(i0: Int, i1: Int, i2: Int, i3: Int, i4: Int, i5: Int, range: UnboundedRange)
  -> DynamicGraph.Tensor<Element>


                    
                    
                    subscript(_:_:_:_:_:_:_:_:)

Declaration

Swift

@inlinable
public subscript(i0: Int, i1: Int, i2: Int, i3: Int, i4: Int, i5: Int, i6: Int,
  range: UnboundedRange
)
  -> DynamicGraph.Tensor<Element>


                    
                    
                    transposed(_:_:streamContext:)

Transpose from axisA to axisB.

Declaration

Swift

public func transposed(_ axisA: Int, _ axisB: Int, streamContext: StreamContext? = nil)
  -> DynamicGraph.Tensor<Element>


                    
                    
                    rand(_:streamContext:)

Fill the given tensor with uniform random values.

Declaration

Swift

public func rand(
  _ range: ClosedRange<Float> = 0...1, streamContext: StreamContext? = nil
)


                    
                    
                    randn(std:mean:streamContext:)

Fill the given tensor with normal-distributed random values.

Declaration

Swift

public func randn(
  std: Float = 1, mean: Float = 0, streamContext: StreamContext? = nil
)


                    
                    
                    toGPU(_:streamContext:)

Copy the given tensor to GPU.

Declaration

Swift

public func toGPU(_ ordinal: Int = 0, streamContext: StreamContext? = nil)
  -> DynamicGraph.Tensor<Element>


                    
                    
                    toCPU(streamContext:)

Copy the given tensor to CPU.

Declaration

Swift

public func toCPU(streamContext: StreamContext? = nil) -> DynamicGraph.Tensor<Element>


                    
                    
                    full(_:streamContext:)

Fill the given tensor with a value.

Declaration

Swift

public func full(
  _ value: Float = 0, streamContext: StreamContext? = nil
)


                    
                    
                    lerp(_:to:streamContext:)

Interpolate from this tensor to the other tensor.

Declaration

Swift

public func lerp(
  _ weight: Float, to: DynamicGraph.Tensor<Element>, streamContext: StreamContext? = nil
)


                    
                    
                    clamp(_:streamContext:)

Clamp the given tensor between two values.

Declaration

Swift

public func clamp(_ range: ClosedRange<Float>, streamContext: StreamContext? = nil)


                    
                    
                    clamp(_:streamContext:)

Clamp the given tensor with a lower bound.

Declaration

Swift

public func clamp(_ range: PartialRangeFrom<Float>, streamContext: StreamContext? = nil)


                    
                    
                    clamp(_:streamContext:)

Clamp the given tensor with an upper bound.

Declaration

Swift

public func clamp(_ range: PartialRangeThrough<Float>, streamContext: StreamContext? = nil)


                    
                    
                    detach()

Detach current tensor from the graph. Afterwards, it is always “isConstant” and cannot requiresGrad.

Declaration

Swift

public func detach()


                    
                    
                    clamped(_:streamContext:)

Clamp the given tensor between two values.

Declaration

Swift

public func clamped(_ range: ClosedRange<Float>, streamContext: StreamContext? = nil)
  -> DynamicGraph.Tensor<Element>


                    
                    
                    clamped(_:streamContext:)

Clamp the given tensor with a lower bound.

Declaration

Swift

public func clamped(_ range: PartialRangeFrom<Float>, streamContext: StreamContext? = nil)
  -> DynamicGraph.Tensor<Element>


                    
                    
                    clamped(_:streamContext:)

Clamp the given tensor with an upper bound.

Declaration

Swift

public func clamped(_ range: PartialRangeThrough<Float>, streamContext: StreamContext? = nil)
  -> DynamicGraph.Tensor<Element>


                    
                    
                    reduced(_:axis:streamContext:)

Declaration

Swift

public func reduced(_ op: ReduceOp, axis: [Int], streamContext: StreamContext? = nil)
  -> DynamicGraph.Tensor<Element>


                    
                    
                    scale(by:streamContext:)

Scale the given tensor with a constant inplace.

Declaration

Swift

public func scale(by a: Float, streamContext: StreamContext? = nil)


                    
                    
                    softmax(streamContext:)

Apply softmax activation to the given tensor inplace.

Declaration

Swift

public func softmax(streamContext: StreamContext? = nil)


                    
                    
                    ReLU(streamContext:)

Apply ReLU activation to the given tensor inplace.

Declaration

Swift

public func ReLU(streamContext: StreamContext? = nil)


                    
                    
                    sigmoid(streamContext:)

Apply sigmoid activation to the given tensor inplace.

Declaration

Swift

public func sigmoid(streamContext: StreamContext? = nil)


                    
                    
                    tanh(streamContext:)

Apply tanh activation to the given tensor inplace.

Declaration

Swift

public func tanh(streamContext: StreamContext? = nil)


                    
                    
                    swish(streamContext:)

Apply swish activation to the given tensor inplace.

Declaration

Swift

public func swish(streamContext: StreamContext? = nil)


                    
                    
                    _Element

Declaration

Swift

public typealias _Element = Element


                    
                    
                    ElementNumeric

Declaration

Swift

public typealias ElementNumeric = Element