Contents Menu Expand Light mode Dark mode Auto light/dark mode
Metatensor
Metatensor
  • What is metatensor
  • Installation
  • Core classes
    • Overview
    • Python API reference
      • TensorMap
      • TensorBlock
      • Labels
      • Serialization
      • Data arrays
      • Miscellaneous
    • C++ API reference
      • TensorMap
      • Labels
      • TensorBlock
      • Data arrays
      • Miscellaneous
    • C API reference
      • TensorMap
      • Labels
      • TensorBlock
      • Data arrays
      • Miscellaneous
    • Rust API reference
    • Tutorials
      • First steps with metatensor
      • Handling sparsity
      • Managing gradients
    • Changelog
  • Operations
    • API reference
      • Checks
        • checks_enabled()
        • unsafe_enable_checks()
        • unsafe_disable_checks()
      • Creation operations
        • empty_like()
        • ones_like()
        • zeros_like()
        • random_like()
        • block_from_array()
      • Linear algebra
        • dot()
        • lstsq()
        • solve()
      • Logic function
        • allclose()
        • equal()
        • equal_metadata()
      • Manipulation operations
        • detach()
        • drop_blocks()
        • join()
        • manipulate dimension
        • one_hot()
        • remove_gradients()
        • requires_grad()
        • samples reduction
        • slice()
        • split()
      • Mathematical functions
        • abs()
        • add()
        • divide()
        • multiply()
        • pow()
        • subtract()
      • Set operations
        • unique_metadata()
        • sort()
    • Changelog
  • TorchScript interface
    • API reference
      • TensorMap
      • TensorBlock
      • Labels
      • Serialization
      • TorchScript C++ API reference
        • TensorMap
        • TensorBlock
        • Labels
        • Miscellaneous
    • Changelog
  • Learning utilities
    • API reference
      • Data utilites
      • Neural Network
    • Tutorials
      • Datasets and data loaders
      • Using IndexedDataset
    • Changelog
  • Atomistic applications
    • API reference
      • Systems
      • Models
        • Exporting models
        • Information about models
      • Atomic Simulation Environment (ASE) integration
      • C++ API reference
        • Systems
        • Models
    • Standard model outputs
    • Tutorials
      • Exporting a model
      • Running Molecular Dynamics with ASE
  • Developer documentation
    • Getting started
    • Code organization
    • Version number management
Back to top

API reference#

Note

This is the documentation for metatensor-learn version 0.3.0.dev1. For other versions, look in the following pages:

Version 0.2.1
Version 0.2.0
Version 0.1.0
  • Data utilites
    • Dataset
      • Dataset
      • IndexedDataset
    • Dataloader
      • DataLoader
    • Collating data
      • group()
      • group_and_join()
  • Neural Network
    • Modules
      • ModuleMap
      • Linear
Next
Data utilites
Previous
Learning utilities
Copyright © 2024, Guillaume Fraux, Davide Tisi, Philip Loche, Joseph W. Abbott, Jigyasa Nigam, Chiheb Ben Mahmoud
Made with Sphinx and @pradyunsg's Furo