Collection

extension Collection where Element: DynamicGraph.AnyTensor
extension Collection where Element: DynamicGraph.AnyGroup
extension Collection where Element: Optimizer
extension Collection where Element == Tensor<Float64>
extension Collection where Element == Tensor<Int64>
extension Collection where Element == Tensor<Float32>
extension Collection where Element == Tensor<Int32>
extension Collection where Element == Tensor<Float16>
extension Collection where Element == Tensor<UInt8>

Available where Element: DynamicGraph.AnyTensor

  • Compute gradients from this tensor to the source tensors.

    Declaration

    Swift

    public func backward<S: Sequence>(to tensors: S, streamContext: StreamContext? = nil)
where S.Element: DynamicGraph.AnyTensor

    Parameters

    to

    The source tensors to this tensor.
    streamContext

    The stream context to perform such operation.

  • Compute gradients from this tensor to the source tensor.

    Declaration

    Swift

    public func backward(to tensor: DynamicGraph.AnyTensor, streamContext: StreamContext? = nil)

    Parameters

    to

    The source tensor to this tensor.
    streamContext

    The stream context to perform such operation.

Available where Element: DynamicGraph.AnyGroup

  • Compute gradients from this tensor to the source tensors.

    Declaration

    Swift

    public func backward<S: Sequence>(to tensors: S, streamContext: StreamContext? = nil)
where S.Element: DynamicGraph.AnyGroup

    Parameters

    to

    The source tensors to this tensor.
    streamContext

    The stream context to perform such operation.

  • Compute gradients from this tensor to the source tensor.

    Declaration

    Swift

    public func backward<Group: DynamicGraph.AnyGroup>(
  to tensor: Group, streamContext: StreamContext? = nil
)

    Parameters

    to

    The source tensor to this tensor.
    streamContext

    The stream context to perform such operation.

Available where Element: Optimizer

  • Update parameters together for a collection of optimizers.

    Declaration

    Swift

    public func step(streamContext: StreamContext? = nil)

    Parameters

    streamContext

    The stream context to have update operation on.

Available where Element == Tensor<Float64>

  • Declaration

    Swift

    public func reshaped(
  format: TensorFormat, dimensions: [Int], offset: [Int]? = nil, increments: [Int]? = nil
) -> [Element]

  • Create new tensors pointing to the same memory region but with different sizes.

    Declaration

    Swift

    public func reshaped(
  _ dimensionFormat: TensorDimensionFormat, offset: [Int]? = nil, increments: [Int]? = nil
) -> [Element]

    Parameters

    dimensionFormat

    New format and dimensions for the tensor.
    offset

    Whether offset on each dimensions.
    increments

    The step on each dimensions.

    Return Value

    The new tensors with different format but the same memory content.

Available where Element == Tensor<Int64>

  • Declaration

    Swift

    public func reshaped(
  format: TensorFormat, dimensions: [Int], offset: [Int]? = nil, increments: [Int]? = nil
) -> [Element]

  • Create new tensors pointing to the same memory region but with different sizes.

    Declaration

    Swift

    public func reshaped(
  _ dimensionFormat: TensorDimensionFormat, offset: [Int]? = nil, increments: [Int]? = nil
) -> [Element]

    Parameters

    dimensionFormat

    New format and dimensions for the tensor.
    offset

    Whether offset on each dimensions.
    increments

    The step on each dimensions.

    Return Value

    The new tensors with different format but the same memory content.

Available where Element == Tensor<Float32>

  • Declaration

    Swift

    public func reshaped(
  format: TensorFormat, dimensions: [Int], offset: [Int]? = nil, increments: [Int]? = nil
) -> [Element]

  • Create new tensors pointing to the same memory region but with different sizes.

    Declaration

    Swift

    public func reshaped(
  _ dimensionFormat: TensorDimensionFormat, offset: [Int]? = nil, increments: [Int]? = nil
) -> [Element]

    Parameters

    dimensionFormat

    New format and dimensions for the tensor.
    offset

    Whether offset on each dimensions.
    increments

    The step on each dimensions.

    Return Value

    The new tensors with different format but the same memory content.

Available where Element == Tensor<Int32>

  • Declaration

    Swift

    public func reshaped(
  format: TensorFormat, dimensions: [Int], offset: [Int]? = nil, increments: [Int]? = nil
) -> [Element]

  • Create new tensors pointing to the same memory region but with different sizes.

    Declaration

    Swift

    public func reshaped(
  _ dimensionFormat: TensorDimensionFormat, offset: [Int]? = nil, increments: [Int]? = nil
) -> [Element]

    Parameters

    dimensionFormat

    New format and dimensions for the tensor.
    offset

    Whether offset on each dimensions.
    increments

    The step on each dimensions.

    Return Value

    The new tensors with different format but the same memory content.

Available where Element == Tensor<Float16>

  • Declaration

    Swift

    public func reshaped(
  format: TensorFormat, dimensions: [Int], offset: [Int]? = nil, increments: [Int]? = nil
) -> [Element]

  • Create new tensors pointing to the same memory region but with different sizes.

    Declaration

    Swift

    public func reshaped(
  _ dimensionFormat: TensorDimensionFormat, offset: [Int]? = nil, increments: [Int]? = nil
) -> [Element]

    Parameters

    dimensionFormat

    New format and dimensions for the tensor.
    offset

    Whether offset on each dimensions.
    increments

    The step on each dimensions.

    Return Value

    The new tensors with different format but the same memory content.

Available where Element == Tensor<UInt8>

  • Declaration

    Swift

    public func reshaped(
  format: TensorFormat, dimensions: [Int], offset: [Int]? = nil, increments: [Int]? = nil
) -> [Element]

  • Create new tensors pointing to the same memory region but with different sizes.

    Declaration

    Swift

    public func reshaped(
  _ dimensionFormat: TensorDimensionFormat, offset: [Int]? = nil, increments: [Int]? = nil
) -> [Element]

    Parameters

    dimensionFormat

    New format and dimensions for the tensor.
    offset

    Whether offset on each dimensions.
    increments

    The step on each dimensions.

    Return Value

    The new tensors with different format but the same memory content.