From 6abcc0d60be126e8392b6e7344344a87df9561be Mon Sep 17 00:00:00 2001 From: Wessel Bruinsma Date: Thu, 24 Jun 2021 18:34:48 +0100 Subject: [PATCH] Add description of device argument --- README.md | 3 +++ requirements.txt | 2 +- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index c2be2f7..841db9c 100644 --- a/README.md +++ b/README.md @@ -218,6 +218,9 @@ This section lists all available constants and methods. - `axis` indicates an axis over which the function may perform its action. An axis is always given as a keyword argument. + - + `device` refers to a device on which a tensor can placed, which can + either be a framework-specific type or a string, e.g. `"cpu"`. - `ref` indicates a *reference tensor* from which properties, like its shape and data type, will be used. E.g., `zeros(tensor)` creates a diff --git a/requirements.txt b/requirements.txt index c67d947..08a2995 100644 --- a/requirements.txt +++ b/requirements.txt @@ -7,7 +7,7 @@ pytest-cov coveralls autograd>=1.3 tensorflow>=2 -tensorflow_probability +tensorflow-probability torch jax jaxlib