Contents Menu Expand Light mode Dark mode Auto light/dark mode Skip to content
Metatensor
Metatensor
  • Metatensor’s goals
  • 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
    • Operations and PyTorch
    • API reference
      • 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()
      • Checks
    • Changelog
  • TorchScript backend
    • 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
    • Overview
    • 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.dev83. 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
      • Sequential
      • Linear
      • EquivariantLinear
      • Tanh
      • InvariantTanh
      • ReLU
      • InvariantReLU
      • SiLU
      • InvariantSiLU
      • LayerNorm
      • InvariantLayerNorm
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