import torch x=torch.arange(12) x.shape x.numel() X = x.reshape(3,4) torch.zeros((2,3,4)) torch.ones((2,3,4))
import os