k api to pytorch

PyTorch, a deep learning framework written in python, also has a c++ library, libtorch, which is delivered as a single zip file containing all the necessary Nvidia libraries and routines to build and train neural networks, along with basic linear algebra routines.

This interface links k to the c++ routines, but attempts to follow the line of the python interface, which also drives the design of the c++ library.

• Overview
• Building ktorch
• Configuration
• Quick start
• Data types
• Pointers
• Devices

Tensors

• Tensors
• Complex tensors
• Sparse tensors
• Vectors
• Dictionaries
• Tensor information
• Tensor operations
• Tensor math

Modules

• Modules
• k api modules
• Initializing parameters
• Loss Modules

Training

• Optimizers
• Models
• Training steps
• Weight averaging
• Freeze layers
• Distributed training

Index

• Index

• Search Page