File size: 73 Bytes
5fa1a76
1
This will work regardless of whether you are using PyTorch or Tensorflow.