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 }


                    
                    
                    typeErased

Declaration

Swift

public var typeErased: AnyTensor { get }


                    
                    
                    subscript(_:streamContext:)

Declaration

Swift

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


                    
                    
                    subscript(_:streamContext:)

Declaration

Swift

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


                    
                    
                    subscript(_:streamContext:)

Declaration

Swift

@inlinable
public subscript(range: Range<Int>, streamContext streamContext: StreamContext?)
  -> DynamicGraph.Tensor<Element>


                    
                    
                    subscript(_:_:streamContext:)

Declaration

Swift

@inlinable
public subscript(i0: Int, range: Range<Int>, streamContext streamContext: StreamContext?)
  -> DynamicGraph.Tensor<Element>


                    
                    
                    subscript(_:_:_:streamContext:)

Declaration

Swift

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


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

Declaration

Swift

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


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

Declaration

Swift

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


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

Declaration

Swift

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


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

Declaration

Swift

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


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

Declaration

Swift

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


                    
                    
                    subscript(_:streamContext:)

Declaration

Swift

@inlinable
public subscript(range: UnboundedRange, streamContext streamContext: StreamContext?)
  -> DynamicGraph.Tensor<Element>


                    
                    
                    subscript(_:_:streamContext:)

Declaration

Swift

@inlinable
public subscript(i0: Int, range: UnboundedRange, streamContext streamContext: StreamContext?)
  -> DynamicGraph.Tensor<Element>


                    
                    
                    subscript(_:_:_:streamContext:)

Declaration

Swift

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


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

Declaration

Swift

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


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

Declaration

Swift

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


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

Declaration

Swift

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


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

Declaration

Swift

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


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

Declaration

Swift

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


                    
                    
                    transposed(_:_:streamContext:)

Transpose from axisA to axisB.

Declaration

Swift

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


                    
                    
                    rand(_:streamContext:)

Fill the given tensor with uniform random values.

Declaration

Swift

public func rand(
  _ range: ClosedRange<Float>, streamContext: StreamContext?
)


                    
                    
                    randn(std:mean:streamContext:)

Fill the given tensor with normal-distributed random values.

Declaration

Swift

public func randn(
  std: Float, mean: Float, streamContext: StreamContext?
)


                    
                    
                    toGPU(_:streamContext:)

Copy the given tensor to GPU.

Declaration

Swift

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


                    
                    
                    toCPU(streamContext:)

Copy the given tensor to CPU.

Declaration

Swift

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


                    
                    
                    full(_:streamContext:)

Fill the given tensor with a value.

Declaration

Swift

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


                    
                    
                    lerp(_:to:streamContext:)

Interpolate from this tensor to the other tensor.

Declaration

Swift

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


                    
                    
                    clamp(_:streamContext:)

Clamp the given tensor between two values.

Declaration

Swift

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


                    
                    
                    clamp(_:streamContext:)

Clamp the given tensor with a lower bound.

Declaration

Swift

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


                    
                    
                    clamp(_:streamContext:)

Clamp the given tensor with an upper bound.

Declaration

Swift

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


                    
                    
                    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?)
  -> DynamicGraph.Tensor<Element>


                    
                    
                    clamped(_:streamContext:)

Clamp the given tensor with a lower bound.

Declaration

Swift

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


                    
                    
                    clamped(_:streamContext:)

Clamp the given tensor with an upper bound.

Declaration

Swift

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


                    
                    
                    copied(streamContext:)

Make a copy of the existing tensor.

Declaration

Swift

public func copied(streamContext: StreamContext?)
  -> DynamicGraph.Tensor<Element>


                    
                    
                    contiguous(streamContext:)

Only make a copy of the existing tensor if it is not contiguous in memory.

Declaration

Swift

public func contiguous(streamContext: StreamContext?)
  -> DynamicGraph.Tensor<Element>


                    
                    
                    reduced(_:axis:streamContext:)

Reduce along a given dimension.

Declaration

Swift

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


                    
                    
                    scale(by:streamContext:)

Scale the given tensor with a constant inplace.

Declaration

Swift

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


                    
                    
                    scaled(by:streamContext:)

Scale the given tensor with a constant.

Declaration

Swift

public func scaled(by a: Float, streamContext: StreamContext?) -> Self


                    
                    
                    softmax(streamContext:)

Apply softmax activation to the given tensor inplace.

Declaration

Swift

public func softmax(streamContext: StreamContext?)


                    
                    
                    ReLU(streamContext:)

Apply ReLU activation to the given tensor inplace.

Declaration

Swift

public func ReLU(streamContext: StreamContext?)


                    
                    
                    sigmoid(streamContext:)

Apply sigmoid activation to the given tensor inplace.

Declaration

Swift

public func sigmoid(streamContext: StreamContext?)


                    
                    
                    tanh(streamContext:)

Apply tanh activation to the given tensor inplace.

Declaration

Swift

public func tanh(streamContext: StreamContext?)


                    
                    
                    swish(streamContext:)

Apply swish activation to the given tensor inplace.

Declaration

Swift

public func swish(streamContext: StreamContext?)


                    
                    
                    init(from:streamContext:)

Explicitly do conversion between types.

Declaration

Swift

public convenience init(from input: DynamicGraph.AnyTensor, streamContext: StreamContext? = nil)


                    
                    
                    chunked(_:axis:streamContext:)

Declaration

Swift

public func chunked(_ numberOfChunks: Int, axis: Int, streamContext: StreamContext?)
  -> [DynamicGraph.Tensor<Element>]


                    
                    
                    ElementNumeric

Declaration

Swift

public typealias ElementNumeric = Element