Files
docker-stacks/tests/by_image/pytorch-notebook/units/unit_pytorch.py

6 lines
172 B
Python

# Copyright (c) Jupyter Development Team.
# Distributed under the terms of the Modified BSD License.
import torch
print(torch.tensor([[1.0, 4.0, 7.0], [4.0, 9.0, 11.0]]))