core.datasets.collaters.simple_collater

core.datasets.collaters.simple_collater#

Copyright (c) Meta Platforms, Inc. and affiliates. This source code is licensed under the MIT license found in the LICENSE file in the root directory of this source tree.

Attributes#

T_co

Functions#

data_list_collater(...)

Module Contents#

core.datasets.collaters.simple_collater.T_co#
core.datasets.collaters.simple_collater.data_list_collater(data_list: list[fairchem.core.datasets.atomic_data.AtomicData], otf_graph: bool = False, to_dict: bool = False) → fairchem.core.datasets.atomic_data.AtomicData | dict[str, torch.Tensor]#
Contents