Início / user-1862861

Matt Pitkin's questions

Martin Hope
Matt Pitkin
Asked: 2024-02-22 22:41:10 +0800 CST
  • 6

Recentemente, PyTorch introduziu o tensor aninhado . No entanto, se eu criar um tensor aninhado, por exemplo,

import torch

a = torch.randn(20, 128)
nt = torch.nested.nested_tensor([a, a], dtype=torch.float32)

e então observe seu tipo de classe, ele mostra:

type(nt)
torch.Tensor

ou seja, o tipo de classe é apenas um PyTorch normal Tensor. Então, type(nt) == torch.Tensore isinstance(nt, torch.Tensor)ambos retornarão True.

Então, minha pergunta é: existe uma maneira de diferenciar entre um tensor regular e um tensor aninhado?

Uma maneira que posso pensar é que o sizemétodo para tensores aninhados (atualmente) funciona de maneira diferente daquele para tensores regulares, pois requer um argumento, caso contrário, gera a RuntimeError. Então, uma solução pode ser:

def is_nested_tensor(nt):
    if not isinstance(nt, torch.Tensor):
        return False

    try:
        # try calling size without an argument
        nt.size()
        return False
    except RuntimeError:
        return True

    return False

mas existe algo mais simples que não depende de algo como o sizemétodo não mudar no futuro?

python
Martin Hope
Matt Pitkin
Asked: 2023-09-17 05:21:38 +0800 CST
  • 3

Eu tenho um monte de strings matemáticas inline no estilo LaTeX em Python, por exemplo, eq = "$E = mc^2$"ou eq = "$F = ma$ (N)"que estou escrevendo em uma página da web que usa MathJAX. Portanto, quero converter os $símbolos de abertura e fechamento no estilo MathJAX padrão de abertura \(e fechamento com \), para que meus dois exemplos acima se tornem eqmathjax = "\(E = mc^2\)e eqmathjax = "\(F = ma\) (N)". Observe que uma string pode, na verdade, conter mais de uma equação, cada uma delimitada por seu próprio par de $sinais.

Eu poderia facilmente iterar as substituições conforme necessário, por exemplo:

twoeqns = "$E = mc^2$ and $F = ma$"
i = 0
repl = ["\(", "\)"]
while "$" in twoeqns:
    twoeqns = twoeqns.replace("$", repl[i % 2], 1)
    i += 1

mas tenho certeza de que deve haver uma maneira mais simples de fazer isso, por exemplo, com expressões regulares (sei que poderia ter a solução acima e uma função e usá-la como uma linha). Alguém tem esse método mais simples?

python