element_spec.json

{
  "__qualname__": "builtins:dict",
  "0": [
    [
      "canvas",
      {
        "__qualname__": "etils.enp.array_spec:ArraySpec",
        "shape": {
          "__qualname__": "builtins.tuple",
          "0": [
            4,
            256,
            1
          ]
        },
        "dtype": "int64"
      }
    ],
    [
      "canvas_id",
      {
        "__qualname__": "etils.enp.array_spec:ArraySpec",
        "shape": {
          "__qualname__": "builtins.tuple",
          "0": [
            4,
            256
          ]
        },
        "dtype": "int64"
      }
    ],
    [
      "canvas_mask",
      {
        "__qualname__": "etils.enp.array_spec:ArraySpec",
        "shape": {
          "__qualname__": "builtins.tuple",
          "0": [
            4,
            256
          ]
        },
        "dtype": "bool"
      }
    ],
    [
      "encoder_target",
      {
        "__qualname__": "etils.enp.array_spec:ArraySpec",
        "shape": {
          "__qualname__": "builtins.tuple",
          "0": [
            4,
            512
          ]
        },
        "dtype": "int64"
      }
    ],
    [
      "encoder_target_mask",
      {
        "__qualname__": "etils.enp.array_spec:ArraySpec",
        "shape": {
          "__qualname__": "builtins.tuple",
          "0": [
            4,
            512
          ]
        },
        "dtype": "bool"
      }
    ],
    [
      "prompt",
      {
        "__qualname__": "etils.enp.array_spec:ArraySpec",
        "shape": {
          "__qualname__": "builtins.tuple",
          "0": [
            4,
            256
          ]
        },
        "dtype": "int64"
      }
    ]
  ]
}