class_docstr: A subclass of :class:`~ffcv.fields.Field` supporting
  multi-dimensional fixed size matrices of any torch type.
class_docstr_tok: ["A","subclass","of",":","class",":","`","~ffcv.fields.Field","`","supporting","multi-dimensional","fixed","size","matrices","of","any","torch","type","."]
code: def __init__(self, dtype: ch.dtype, shape: Tuple[int, ...]):
      self.dtype = dtype
      self.shape = shape
      dtype = ch.zeros(0, dtype=dtype).numpy().dtype
  
      super().__init__(dtype, shape)
  
imports: from typing import Callable, TYPE_CHECKING, Tuple, Type
  import warnings
  import json
  from dataclasses import replace
  import numpy
  import torch
  from .base import Field, ARG_TYPE
  from ..pipeline.operation import Operation
  from ..pipeline.state import State
  from ..pipeline.compiler import Compiler
  from ..pipeline.allocation_query import AllocationQuery
  from ..libffcv import memcpy
label_desc: Use image node_id 1 to create a new TorchTensorField object from inherited base classes: NDArrayField with example: obj = TorchTensorField(dtype, shape)