import torch as tf