package torch

You can search for identifiers within the package.

in-package search v0.2.0

PyTorch bindings for OCaml

Install

md5=7a712ae0e8c7f5452f628377d80a5bb4

sha512=22314b655bc6b5e5c970cbab8d132eae36ee0b8fb0a96b63727899442eb70fe00bd1895d7cc718a85b58bc2b2b4ea6820fa288a19346f095e5de18f7e47c2d02

1
2
3
4
5
6
7
8
9
type t =
  | None
  | Elementwise_mean
  | Sum

let to_int = function
  | None -> 0
  | Elementwise_mean -> 1
  | Sum -> 2