PyTorch Tensor Example in Colab - Google-colab Typing CST Test