Cảnh báo: Dự án này không được chấp nhận. Swift cho TensorFlow là một thử nghiệm trong nền tảng thế hệ tiếp theo dành cho máy học, kết hợp nghiên cứu mới nhất về học máy, trình biên dịch, lập trình phân biệt, thiết kế hệ thống, v.v. Nó được lưu trữ vào tháng 2 năm 2021.

Trang này được dịch bởi Cloud Translation API.

Trạng thái Chủ đề

public class ThreadState<Model: Layer, Opt: Optimizer>
where
  Opt.Model == Model, Opt.Scalar == Float, Model.Input == Tensor<Float>,
  Model.Output == Tensor<Float>,
  Model.TangentVector.VectorSpaceScalar == Float

Trạng thái của vòng lặp huấn luyện trên thiết bị.

sự phân loại
Tuyên ngôn
public var classifier: Model
trình tối ưu hóa
Tuyên ngôn
public var optimizer: Opt

init(model:optimizer:id:devices:useAutomaticMixedPrecision:)

Tuyên ngôn

public init(
  model: Model, optimizer: Opt, id: Int, devices: [Device], useAutomaticMixedPrecision: Bool
)

run(train:test:crossReplicaSumDevices:scheduleLearningRate:lossFunction:)

Tuyên ngôn

public func run<Dataset: Sequence>(
  train: Dataset, test: Dataset, crossReplicaSumDevices: [Device]? = nil,
  scheduleLearningRate: (Opt) -> Void = { _ in },
  lossFunction: @differentiable (Tensor<Float>, @noDerivative Tensor<Int32>) -> Tensor<Float> =
    _defaultLossFunction
)
  -> () -> (train: HostStatistics, test: HostStatistics)
where Dataset.Iterator.Element == (x: Tensor<Float>, y: Tensor<Int32>)