Contents Menu Expand Light mode Dark mode Auto light/dark, in light mode Auto light/dark, in dark mode Skip to content
KOSMOS 0.3.1 Documentation
KOSMOS 0.3.1 Documentation

User Guide

  • Getting started
  • Examples
    • Machine Learning Examples
      • Centralized Learning
        • Centralized Learning – Variational Quantum Circuits
        • Centralized Learning – Neural Networks
      • Federated Learning
        • Federated Learning – Variational Quantum Circuits
        • Federated Learning – Neural Networks
    • Routing Examples

API Documentation

  • kosmos.ml
    • kosmos.ml.config
      • kosmos.ml.config.factories
        • kosmos.ml.config.factories.encoding
        • kosmos.ml.config.factories.loss
        • kosmos.ml.config.factories.lr_scheduler
        • kosmos.ml.config.factories.model
        • kosmos.ml.config.factories.optimizer
      • kosmos.ml.config.sl_train
    • kosmos.ml.datasets
      • kosmos.ml.datasets.banknote_dataset
      • kosmos.ml.datasets.dataset
      • kosmos.ml.datasets.digits_dataset
      • kosmos.ml.datasets.fashion_dataset
      • kosmos.ml.datasets.income_dataset
      • kosmos.ml.datasets.iris_dataset
      • kosmos.ml.datasets.mnist_dataset
      • kosmos.ml.datasets.qsar_dataset
      • kosmos.ml.datasets.wdbc_dataset
      • kosmos.ml.datasets.wine_dataset
    • kosmos.ml.fl
      • kosmos.ml.fl.fl_client
      • kosmos.ml.fl.fl_manager
      • kosmos.ml.fl.fl_server
    • kosmos.ml.models
      • kosmos.ml.models.vqc
        • kosmos.ml.models.vqc.circuit
        • kosmos.ml.models.vqc.encoding
        • kosmos.ml.models.vqc.vqc
      • kosmos.ml.models.model
      • kosmos.ml.models.neural_network
    • kosmos.ml.cl_manager
    • kosmos.ml.dataloader
    • kosmos.ml.sl_metrics
    • kosmos.ml.sl_result
    • kosmos.ml.sl_trainer
    • kosmos.ml.typing
  • kosmos.protocols
    • kosmos.protocols.config
      • kosmos.protocols.config.protocol
    • kosmos.protocols.routing
      • kosmos.protocols.routing.dijkstra_routing
      • kosmos.protocols.routing.path
      • kosmos.protocols.routing.routing
    • kosmos.protocols.eg_protocol
    • kosmos.protocols.protocol
    • kosmos.protocols.protocol_result
    • kosmos.protocols.status
  • kosmos.quantum_logic
    • kosmos.quantum_logic.entanglement_link
    • kosmos.quantum_logic.quantum_register_manager
    • kosmos.quantum_logic.quantum_state
    • kosmos.quantum_logic.qubit
    • kosmos.quantum_logic.typing
  • kosmos.simulator
    • kosmos.simulator.fl_simulator
    • kosmos.simulator.simulator
  • kosmos.topology
    • kosmos.topology.link
    • kosmos.topology.net
    • kosmos.topology.node
    • kosmos.topology.typing
  • kosmos.utils
    • kosmos.utils.rng
Back to top
View this page

kosmos.ml.typing¶

Module Attributes¶

type TensorMapping = Callable[[torch.Tensor], torch.Tensor]¶
Next
kosmos.protocols
Previous
kosmos.ml.sl_trainer
Copyright © 2025, AIGNOSCO GmbH
Made with Sphinx and @pradyunsg's Furo
On this page
  • kosmos.ml.typing
    • Module Attributes
      • TensorMapping